美文网首页
面试题34. 二叉树中和为某一值的路径

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

作者: 周英杰Anita | 来源:发表于2020-03-01 17:17 被阅读0次

    题目描述:

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

    给定如下二叉树,以及目标和 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<List<Integer>> ans;
        public List<List<Integer>> pathSum(TreeNode root, int sum) {
            ans = new ArrayList<>();
            if(root == null)
            {
                return ans;
            }
            findPath(new ArrayList<Integer>(), root , sum, 0);
            return ans;
        }
        public void findPath(List<Integer> 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. 二叉树中和为某一值的路径

          本文链接:https://www.haomeiwen.com/subject/cpgckhtx.html