找二叉树中和为某一值的路径
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
这题磨耗了很多时间,首先要理解题目意思,题目指从根节点到叶子节点路径的和为某一个目标值,所以路径中的最后一个节点一定是叶子节点。所以只要遍历所有到叶子节点的路径,再看路径的节点和是否为目标值即可解答该题。
解答一:
/** * 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); } } }
该解答方法,没有合理使用最初的列表,为每一个节点复制一个列表,以达到路径唯一的特性,但是极大损耗内存,效率也不高。
总结:
我们在解决问题时候还是要看清题意,合理使用数据结构,才能达到比较高效的解决方法