剑指Offer-21——二叉树中和为某一值的路径

题目描述

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

解析

如果target为10，那么符合的路径就依次是1,2,7和1,4,5.
这里肯定是要用递归的方式进行遍历。

• 首先每次遍历的作为“根”的结点,都直接放入路径中，因为每一条路径都会经过根结点。
• 然后再将target更新，减去放入路径存储表中的值。
• 然后在已经target的值进行判断，是否是正确的符合要求的路径。
  （1）target<0，说明该条路径不符合要求。
  （2）target==0,并且该结点的左右结点都没有余下的左右子结点（说明是叶子结点），因此该路径就应该是符合要求的。将此路径加入最终的结果立链表中。
  （3）target>0，那么就需要将其左右子结点作为根结点进行循环，重新开始。

代码

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 {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target){
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> path = new ArrayList<Integer>();
        if(root==null) return list;
         find(list,path,target,root);
        return list;
    }
    public void find(ArrayList<ArrayList<Integer>> list,ArrayList<Integer> path,
                     int target,TreeNode root){
        if(root==null)return;
        //首先将节点加入
        path.add(root.val);
        //更新target的值
        target = target - root.val;
        //target<0，说明不是该路径，停止递归
        if(target<0) return;
        //当target==0时，
        if(target==0&&root.left==null&&root.right==null){
            //说明该路径path满足要求，将其加入到list中
            list.add(path);
        }
        //当上面两种情况都不满足，就重新找路径，就继续递归
        find(list,new ArrayList<Integer> (path),target,root.left);
        find(list,new ArrayList<Integer> (path),target,root.right);
    }
}