输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
这题磨耗了很多时间,首先要理解题目意思,题目指从根节点到叶子节点路径的和为某一个目标值,所以路径中的最后一个节点一定是叶子节点。所以只要遍历所有到叶子节点的路径,再看路径的节点和是否为目标值即可解答该题。
解答一:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> pathSum(TreeNode root, int sum) {
List<List<Integer>> result = new ArrayList();
if(root!=null){
List<Integer> list = new ArrayList();
depthVisit(root,0,list,result,sum);
}
return result;
}
void depthVisit(TreeNode root, int sum,List<Integer> list,List<List<Integer>> result,int target){
if(root!=null){
list.add(root.val);
if( (sum+root.val)==target&&root.left==null&&root.right==null){
List<Integer> tempList = new ArrayList();
for(int i=0;i<list.size();i++){
tempList.add(list.get(i));
}
result.add(tempList);
}else{
depthVisit(root.left,sum+root.val,list,result,target);
depthVisit(root.right,sum+root.val,list,result,target);
}
list.remove(list.size()-1);
}
}
}
执行的效率如下:
该解答方法,合理使用一个列表来记录路径,全局只使用一个列表
解答二:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> pathSum(TreeNode root, int sum) {
List<List<Integer>> result = new ArrayList();
if(root!=null){
List<Integer> list = new ArrayList();
depthVisit(root,0,list,result,sum);
}
return result;
}
void depthVisit(TreeNode root, int sum,List<Integer> list,List<List<Integer>> result,int target){
if(root!=null){
if( (sum+root.val)!=target){
list.add(root.val);
}else{
list.add(root.val);
if(root.left==null&&root.right==null){
List<Integer> tempList = new ArrayList();
for(int i=0;i<list.size();i++){
tempList.add(list.get(i));
}
result.add(tempList);
}
}
List<Integer> tempListLeft = new ArrayList();
for(int i=0;i<list.size();i++){
tempListLeft.add(list.get(i));
}
List<Integer> tempListRight = new ArrayList();
for(int i=0;i<list.size();i++){
tempListRight.add(list.get(i));
}
depthVisit(root.left,sum+root.val,tempListLeft,result,target);
depthVisit(root.right,sum+root.val,tempListRight,result,target);
}
}
}
该解答方法,没有合理使用最初的列表,为每一个节点复制一个列表,以达到路径唯一的特性,但是极大损耗内存,效率也不高。
总结:
我们在解决问题时候还是要看清题意,合理使用数据结构,才能达到比较高效的解决方法