剑指offer-二叉树中和为某一值的路径

题目描述

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

 

 

 

题目链接：

https://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca?tpId=13&tqId=11177&rp=1&ru=/activity/oj&qru=/ta/coding-interviews/question-ranking

 

 

 

分析：

遍历全部路径，只有在叶子节点判断是否满足路径和与目标和是否相等。

 

注意点：

1、遍历到该节点，将该节点加入路径和当前总和；同时，在该节点的左右子树遍历完时，需要将该节点的值在总和中减去，并从路径中删除。

2、ArrayList指向一个地址，ArrayList<ArrayList<>>每一项代表一个地址，添加时，需新建一个ArrayList。

 

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    ArrayList<ArrayList<Integer>> paths =new ArrayList<>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        findPath(root,target,0,new ArrayList<>());
        return paths;
    }
    
    public void findPath(TreeNode cur,int target,int sum,ArrayList<Integer> path){
        //剪枝，当前和已大于目标和时直接返回
        if(cur == null || sum >= target){
            return;
        }
        //走过该节点，所以添加到路径中，并加上该节点的值
        path.add(cur.val);
        sum+=cur.val;
        //判断是否是叶子节点，是则判断是否满足和，否则不做操作
        if(cur.left == null && cur.right == null){
            if(target == sum){
                ArrayList<Integer> tmp =new ArrayList<>();
                for(int i =0 ;i<path.size();i++){
                    tmp.add(path.get(i));
                }
                paths.add(tmp);
            }
        }
        //遍历左节点
        if(cur.left != null){
            findPath(cur.left,target,sum,path);
        }
        //遍历右节点
        if(cur.right != null){
            findPath(cur.right,target,sum,path);
        }
        //和减去该节点的值，路径上去掉该节点
        sum-=cur.val;
        path.remove(path.size()-1);
    }
}