面试题34. 二叉树中和为某一值的路径

题目描述:

输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:

给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

提示:

节点总数 <= 10000

Java解法(递归):

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private List> ans;
    public List> pathSum(TreeNode root, int sum) {
        ans = new ArrayList<>();
        if(root == null)
        {
            return ans;
        }
        findPath(new ArrayList(), root , sum, 0);
        return ans;
    }
    public void findPath(List path, TreeNode root, int sum, int curSum)
    {
        curSum += root.val;
        path.add(root.val);
        if(curSum == sum && root.left == null && root.right == null)
        {
            ans.add(new ArrayList(path));
        }
        if(root.left != null){
            findPath(path, root.left, sum, curSum);
            path.remove(path.size() - 1);
        }
        if(root.right != null){
            findPath(path, root.right, sum, curSum);
            path.remove(path.size() - 1);
        }
    }
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof

你可能感兴趣的:(面试题34. 二叉树中和为某一值的路径)