题目:输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路:
采用一种“减法”思想,当检查一棵树从根到叶子节点形成的路径的和是否为target时,先将当前根节点的值 root.val 加入path, 然后检查它的左子树(若非空),看从左子树的根到叶子节点形成的路径的和是否为 target - root.val (递归), 然后同样的道理去递归检查右子树(若非空),这便是大致的思路。
- 但这道题麻烦的一点是,它要求记录下所有符合标准的路径,这便用到了dfs的特性。
-
但又来了一件麻烦事,先序遍历便是先左后右。检查完左子树后,会对path就行修改,再去查找右子树,如何将path恢复到之前未进行左子树检查的状态?
-
比较好的做法是将path设为全局的,然后dfs的过程便是先序遍历的过程,一旦遍历到叶子结点,便将path最后的节点移除掉,这样在递归一层一层进行的时候将值添加进path,在递归返回的过程中将path最末尾的元素一个一个移除。这样便依靠递归的特性完成了路径的恢复。
package jianzhi_offer;
import java.util.ArrayList;
public class P34_Solution {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> path = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root == null)
return res;
findPath(root,target);
return res;
}
public void findPath(TreeNode root, int target) {
path.add(root.val);
if(root.val == target && root.left == null && root.right == null)
res.add(new ArrayList(path));
if(root.left != null) {
findPath(root.left, target - root.val);
}
if(root.right != null) {
findPath(root.right,target - root.val);
}
path.remove(path.size() - 1);
return;
}
public static void main(String[] args) {
// TODO Auto-generated method stub
}
}