二叉树中和为某一值的路径
-
题目:输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
-
思路:
1.当用前序遍历的方式访问到某一节点时,把该节点添加到路径上,并累加该节点的值;
(前中后序遍历即 将要进行的操作放在对应的顺序,然后使用递归)
2.如果该节点为叶节点,并且路径中节点值的和刚好等于目标值,则当前路径符合要求;
3.如果该节点不是叶节点,则递归访问左子节点、右子节点;
4.在函数退出之前,要在路径上删除当前节点并在路径累加和中减去当前节点的值(以确保返回父节点时路径刚好是从根节点到父节点)
5.对于结果list要进行排序,通过自定义比较器,使其按照存放的list的长度逆序排序。 -
代码
import java.util.ArrayList;
import java.util.*;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
if(root==null){
return result;
}
Stack<TreeNode> path = new Stack<TreeNode>();
int sum = 0;
process(root,target,sum,path,result);
//匿名内部类自定义比较器,按照list中存放list的长度逆序排序
Collections.sort(result, new Comparator<ArrayList<Integer>>() {
@Override
public int compare(ArrayList<Integer> o1, ArrayList<Integer> o2) {
return o2.size()-o1.size();
}
});
return result;
}
public void process(TreeNode root,int target,int sum,Stack<TreeNode> path,ArrayList result){
//将当前节点添加到路径中
sum+=root.val;
path.push(root);
ArrayList<Integer> pathList = new ArrayList<Integer>();
//如果是叶节点并且路径和等于目标,则将路径添加到列表中
if(root.left==null&&root.right==null&&sum==target){
for(TreeNode node:path){
pathList.add(node.val);
}
result.add(pathList);
}
//如果不是叶节点,继续访问它的子节点
if(root.left!=null){
process(root.left,target,sum,path,result);
}
if(root.right!=null){
process(root.right,target,sum,path,result);
}
//返回父节点之前,在路径上删除当前节点
path.pop();
}
}