题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
路径总是从根节点到达叶结点的,因此要先遍历根节点!只有先序遍历符合这个特点。访问头结点,将其加入路径中,计算当前的数值是否合适,如果数值合适,并且当前结点的左右孩子为null,那么其将成为一个路径结果!否则,从它的左右孩子中继续寻找,当前结点访问结束后,应该删除当前的结点,让它返回到其父节点继续进行!
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>> list = new ArrayList();
ArrayList<Integer> arr = new ArrayList();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root == null)
return list;
arr.add(root.val);
target = target - root.val;
if(target == 0 && root.left == null && root.right == null){
list.add(new ArrayList(arr));
}
FindPath(root.left,target);
FindPath(root.right,target);
arr.remove(arr.size() - 1);
return list;
}
}
(1)list.add(new ArrayList(arr));
这里list增加arr的话,那么就会是一个引用,由于arr是全局的,所以之后的操作都会改变它,保存的内容会发生变化,所以new一个新的!
(2)最后要删除当前结点!
(3)以下面代码为例:
TreeNode node1 = new TreeNode(10);
TreeNode node2 = new TreeNode(5);
TreeNode node3 = new TreeNode(12);
TreeNode node4 = new TreeNode(4);
TreeNode node5 = new TreeNode(7);
node1.left = node2;
node1.right = node3;
node2.left = node4;
node2.right = node5;
ArrayList<ArrayList<Integer>> result = FindPath(node1, 22);
首先,访问头结点10,加入arr中,target = 12,不符合路径条件!对其左右孩子继续进行!
对其左孩子:
访问结点5,加入arr中,target = 7,不符合条件!对其左右孩子继续!
对其左孩子:
访问结点4,加入arr中,target = 3,不符合条件(值不符合,但左右孩子为null)!对其左右孩子继续!
对其左孩子:
结点为null,返回list(为空)!即if(root == null) return list;
对其右孩子:
结点为null,返回list(为空)!
结点4的左右孩子访问结束,下面要从arr中删除当前节点(4),回到父节点5!
访问结点5的右孩子:
访问结点7,加入arr中,target = 0,符合条件,将当前arr的值重新声明并加入最后的路径列表!
访问结点7的左孩子:
结点为null,返回list(有一个元素)!
访问结点7的右孩子:
结点为null,返回list(有一个元素)!
结点7的左右孩子访问结束,删除arr中的当前结点(7)!返回父节点5
结点5的左右孩子访问结束,删除arr中的当前节点(5),返回父节点(10)!
访问结点10的右孩子:
访问结点12,加入arr中,target = 0,符合条件,将当前arr的值重新声明并加入最后的路径列表list!
访问结点12的左孩子:
结点为null,返回list(有两个元素了)!
访问结点12的右孩子:
结点为null,返回list(有两个元素了)!
结点12的左右孩子访问结束,删除arr中的当前结点(12)!返回父节点10!
结点10的左右孩子访问结束,从arr中删除当前结点10!返回最后的list。
参考:
http://blog.sina.com.cn/s/blog_a1ce3d4b0102wjal.html
https://www.cnblogs.com/lfeng1205/p/6853918.html?utm_source=itdadao&utm_medium=referral
还应该有非递归的写法,等掌握这个之后在了解吧