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

题目

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

思路

使用前序遍历,并统计当前路径的sum。
当某个节点已经不在路径内,需要回溯,将这个节点出栈,并从sum中减去对应的值。

实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
        List<List<Integer>> res = new ArrayList<>();

        public List<List<Integer>> pathSum(TreeNode root, int sum) {
            List<Integer> path = new ArrayList<>();
            dfs(root, sum, path);
            return res;
        }
        
        public void dfs(TreeNode root, int sum, List<Integer> path){
            
            if(root==null) return;

            path.add(root.val);
            
            if(root.left==null && root.right==null )
                if(root.val==sum)
                    res.add(new ArrayList<Integer>(path));
            
            if(root.left!=null) 
                dfs(root.left,sum-root.val,path);
            
            if(root.right!=null) 
                dfs(root.right,sum-root.val,path);
            
            path.remove(path.size()-1);
            
        }
}


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容