首页 > 其他 > 详细

[Algo] 141. Binary Tree Path Sum To Target III

时间:2020-03-05 23:08:54      阅读:83      评论:0      收藏:0      [点我收藏+]

Given a binary tree in which each node contains an integer number. Determine if there exists a path (the path can only be from one node to itself or to any of its descendants), the sum of the numbers on the path is the given target number.

Examples

    5

  /    \

2      11

     /    \

    6     14

  /

 3

If target = 17, There exists a path 11 + 6, the sum of the path is target.

If target = 20, There exists a path 11 + 6 + 3, the sum of the path is target.

If target = 10, There does not exist any paths sum of which is target.

If target = 11, There exists a path only containing the node 11.

How is the binary tree represented?

We use the level order traversal sequence with a special symbol "#" denoting the null node.

 

/**
 * public class TreeNode {
 *   public int key;
 *   public TreeNode left;
 *   public TreeNode right;
 *   public TreeNode(int key) {
 *     this.key = key;
 *   }
 * }
 */
public class Solution {
  public boolean exist(TreeNode root, int target) {
    // Write your solution here
    Set<Integer> set = new HashSet<>();
    set.add(0);
    return helper(root, set, target, 0);
  }

  private boolean helper(TreeNode root, Set<Integer> set, int target, int prefixSum) {
    if (root == null) {
      return false;
    }
    int curSum = root.key + prefixSum;
    if (set.contains(curSum - target)) {
      return true;
    }

    boolean needRemoval = set.add(curSum);
    boolean left = helper(root.left, set, target, curSum);
    boolean right = helper(root.right, set, target, curSum);
    // it is possible to remove previous presum here, 
    // such as 0, pathsum = 0, cannot check pathsum from root after here
    if (needRemoval) {
      set.remove(curSum);
    }
    return left || right;
  } 
}

 

[Algo] 141. Binary Tree Path Sum To Target III

原文:https://www.cnblogs.com/xuanlu/p/12423533.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!