112. Path Sum

[miyataka]

https://leetcode.com/problems/path-sum/

[h-masder]

conditional ORが短絡評価をしてくれるので、それを使うことで、(全部見ることなく)発見したタイミングでreturn できます。

https://go.dev/ref/spec?utm_source=chatgpt.com#Logical_operators

func hasPathSumFunc(root *TreeNode, targetSum, pathSum int) bool {
    if root == nil {
        return false
    }
    // 停止条件
    if root.Left == nil && root.Right == nil {
        return root.Val + pathSum == targetSum
    }
    pathSum += root.Val
    return hasPathSumFunc(root.Left, targetSum, pathSum) || hasPathSumFunc(root.Right, targetSum, pathSum)
}

[miyataka]

たしかに．自分のコードだと短絡評価を活用できていないですね．
leetcodeの判定が0msだったので，ちょっと満足してしまっていたかもしれないですね．

コメントありがとうございます

[h-masder]

関数名がhasPathSumなので、true/falseを返していいのかなと思いました。
また、breakせずともその場でreturnしてもいいのかなと思いました。

    for len(toVisit) > 0 {
        v := toVisit[len(toVisit)-1]
        toVisit = toVisit[:len(toVisit)-1]
        node, pathSum := v.node, v.pathSum

        if node == nil {
            continue
        }
        nextPathSum := pathSum + node.Val
        if node.Left == nil && node.Right == nil {
            if nextPathSum == targetSum {
                return true
            }
        }
        toVisit = append(toVisit, nodeAndPathSum{node: node.Left, pathSum: nextPathSum})
        toVisit = append(toVisit, nodeAndPathSum{node: node.Right, pathSum: nextPathSum})
    }
    return false
}

[miyataka]

関数名がhasPathSumなので、true/falseを返していいのかなと思いました。

これはそれもアリだなと思います．
書いた時にはreturnする場所はあまり増やしたくないなと感じてました．
一方で普段early returnも有用だと考えてるので，なんか個人的に使い分けをもうちょい明確にしたい部分だなと気づきました．

コメントありがとうございます

[h-masder]

全体的に読みやすかったです。