JAva判断ツリーパス総和

3936 ワード

ツリーとターゲット和を指定し、ツリーにルートノードからリーフノードへのパスが存在するかどうかを判断します.このパスのすべてのノード値がターゲット和に加算されます.
説明:リーフノードとは、サブノードがないノードを指します.
例:
次のような二叉木とターゲットとsum=22が与えられ、
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

ターゲットおよび22のルートノードからリーフノードへのパスtrueが存在するため、5->4->11->2を返す.
解析:ルートノード(root)からリーフノードへのパス加算がターゲットと(sum)->ルートノード(root)の左右ノードに残りのターゲットと等しいパス加算があるかどうか(sum-root.val)
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
       public boolean hasPathSum(TreeNode root, int sum) {

        TreeNode left = root.left;
        TreeNode right = root.right;
        //              ,      
        if (left == null && right == null) return sum == root.val;
        boolean b1 = false, b2 = false;
        //                
        if (left != null) b1 = hasPathSum(left, sum - root.val);
        //                
        if (right != null) b2 = hasPathSum(right, sum - root.val);
        //             
        return b1 || b2;
    }
}
  • ルートノードが空であると判断する必要がある場合、必要な先頭に1行の判断if(root == null) return false;
  • を追加する.