题目:二叉树中和为某一值的路径
描述:输入一颗二叉树的根节点和一个整数,按字典序打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路:对二叉树进行深度优先遍历,每遍历一个节点都用和减去这个节点的值,遍历到叶子节点,要是和为零,则这条遍历路径就是目标路径,否则返回上一层节点,本题中,输入{10,5,12,4,7},22,如图:
10→5→4,10+5+4≠22,这条路径不满足,不是目标路径,10→5→7,10+5+7=22,路径满足要求,是目标路径,10→12,10+12=22,路径满足要求,是目标路径。
Java代码如下所示:
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 {
private ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>();
private ArrayList<Integer> list = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root == null){
return listAll;
}
list.add(root.val);
target -=root.val;
if(target == 0 && root.left == null && root.right == null){
listAll.add(new ArrayList<Integer>(list));
}
if(root.left!=null){
FindPath(root.left,target);
}
if(root.right !=null){
FindPath(root.right,target);
}
list.remove(list.size()-1);
return listAll;
}
}
时间复杂度:需要遍历树中的元素,时间复杂度为0(N)。



京公网安备 11010502036488号