描述
输入一颗二叉树的根节点root和一个整数expectNumber,找出二叉树中结点值的和为expectNumber的所有路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点,不能从子节点到父节点
4.总节点数目为n
如二叉树root为{10,5,12,4,7},expectNumber为22
则合法路径有[[10,5,7],[10,12]]
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
//定义两个空间,一个做返回值,一个存储数
private ArrayList< ArrayList<Integer>> list = new ArrayList<>();
private Stack<Integer> path = new Stack<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int expectNumber) {
if(root == null) return list;
//每个元素都压入栈
path.push(root.val);
//走一个节点减一下
expectNumber -= root.val;
//判断是否为叶子节点并expectNumber==0
if(expectNumber == 0 && root.left == null&&root.right==null){
list.add(new ArrayList<Integer>(path));
}
//递归
FindPath(root.left,expectNumber);
FindPath(root.right,expectNumber);
//无论是不是都把叶子节点删除,进行下一个判断
path.pop();
return list;
}
}