112. Path Sum

[seal-azarashi]

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

[fhiyo]

いいと思います！

[liquo-rice]

        if (isLeafNode) {
            return  root.val == targetSum;
        }

[seal-azarashi]

step 4 に修正版を書きました

[liquo-rice]

上と同じく、葉ノードならここでリターンするのが自然な気がします。

[seal-azarashi]

こちらの場合は currentPathSum == targetSum が false の場合は処理を継続しなければならないので、こんな感じでしょうか

            if (isLeafNode) {
                if (currentPathSum == targetSum) {
                    return true;
                }
                
                continue;
            }

[liquo-rice]

そうです！

[seal-azarashi]

ありがとうございます！こちらも step 4 に記載しました

[oda]

私はこれを変数には置かない気がします。ただ趣味の範囲です。

[seal-azarashi]

返信遅くなりました。
確認するのに時間が空いたからか、改めて見ると冗長に感じたので、変数に置かないように step 4 を修正しました。

[oda]

Java は最近(Tiger 以降)ほとんど書いていないのですが、record は null 入らないんでしたっけ。スタックには null も入るようにして、チェックを pop してから行うのも一つかと思いました。

[seal-azarashi]

返信遅くなりました。
Null が入らないのは ArrayDeque の方になります。 Record はイミュータブルですが null を代入することは可能です。
確かに、別のクラスを使って、チェックを pop してから行うのも一つですが、経験上 null を許容しない設計になっているクラス (メンバ etc.) ないしコレクションの方が扱いやすかったので、こちらをあえて選んでいました。

[oda]

あれ、ArrayDeque は、null が入らないのはいいですが、null を要素に持つ Record も入らないということですか?
それとも、文法上はTreeNode node = nodeWithPathSum.node;の直後に null check をして、あと3箇所のチェックを消すというのも選択として可能ということですか。

[seal-azarashi]

あ、すいません読み違えてました。Null であるフィールドを持つ record は許容されます (null だとだめなのは record のインスタンスでした)。ですので、スタックには null も入るようにして、チェックを pop してから行うのは可能です。

こんな実装になります (step 4 に記載しました)

class Solution {
    private record NodeWithPathSum(TreeNode node, int pathSum) {};

    public boolean hasPathSum(TreeNode root, int targetSum) {
        if (root == null) {
            return false;
        }

        Deque<NodeWithPathSum> nodeStack = new ArrayDeque<>();
        nodeStack.push(new NodeWithPathSum(root, 0));
        while (!nodeStack.isEmpty()) {
            NodeWithPathSum nodeWithPathSum = nodeStack.pop();
            TreeNode node = nodeWithPathSum.node;

            if (node == null) {
                continue;
            }

            int currentPathSum = node.val + nodeWithPathSum.pathSum;
            if (node.left == null && node.right == null) {
                if (currentPathSum == targetSum) {
                    return true;
                }
                
                continue;
            }
            nodeStack.push(new NodeWithPathSum(node.right, currentPathSum));
            nodeStack.push(new NodeWithPathSum(node.left, currentPathSum));
        }
        return false;
    }
}

[seal-azarashi]

こちらの方が見通しが良く感じました。