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

Posted by     DH on May 9, 2017

题目

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

分析

路劲指的是从根节点开始到叶子节点的一个通路,所以首先必须访问根节点,在几种遍历树的方法中,前序遍历的顺序是根->左->右。 所以,我们采用前序遍历,首先访问根节点,然后访问根节点的左孩子。这个时候,我们的当前节点就是左孩子,最开始的根节点就无法访问, 所以需要用一个栈进行保存,方便回退。这里我选用的是ArrayList onePath = new ArrayList(); 作为存储。 当我们访问到叶子节点时,入股所有节点的值的和不等于target,那么久回退到上一个节点,去访问他的子节点。

代码

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<ArrayList<Integer>> allPath = new ArrayList<ArrayList<Integer>>();
    ArrayList<Integer> onePath = new ArrayList<Integer>();

    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root == null)
            return allPath;

        onePath.add(root.val);
        target = target - root.val;
        if(target == 0 && root.left == null && root.right == null){
            allPath.add(new ArrayList<Integer>(onePath));
        }

        FindPath(root.left,target);
        FindPath(root.right,target);

        onePath.remove(onePath.size() - 1);


        return allPath;
    }
}