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

Posted 2020-08-10 萝卜er

题目描述

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

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>> al1=new ArrayList<ArrayList<Integer>>();
    ArrayList<Integer> al2=new ArrayList<Integer>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root==null){
            return al1;
        }
        test(root,0,target,al1,al2);
        return al1;
        
    }
    public void test(TreeNode root,int sum,int target,ArrayList<ArrayList<Integer>> al1,ArrayList<Integer> al2){
            al2.add(root.val);
            if(sum+root.val==target&&root.left==null&&root.left==null){
                al1.add(new ArrayList<Integer>(al2));
            }
            if(root.left!=null) test(root.left,sum+root.val,target,al1,al2);
            if(root.right!=null) test(root.right,sum+root.val,target,al1,al2);//限制了执行该函数的节点都不是空的
            al2.remove(al2.size()-1);

        }
    }

这个可以看做是一个深度优先遍历算法。

因为限制了节点不能为空，所以第一步是把当前的节点的值插入到al2中，接下来判断节点是不是叶子节点，如果是叶子节点，而且当前sum+node.val等于target，则把al2插入到al1中

如果不满足上述条件，那么就把他的左右子树分别放进去，递归执行，然后当前执行完了之后，本来是要返回给上一层的，所以要把当前层插入进去的数据去掉