题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
编程思路
使用前序遍历从根节点到叶结点找到一条路径,若路径之和等于给定的值,则这条路径即为要找的路径,具体实现如下:
若根节点为空或根结点的值大于给定的值,则清空路径;
若根节点的值等于给定的值,则要判断该结点是否是叶子结点,若是,则该路径是要找的路径,若不是,则将保存的结点清空;
若根节点的值小于给定的值,则递归访问它的左右子树。
程序代码(Java语言)
import java.util.ArrayList;
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>> pathlist = new ArrayList<ArrayList<Integer>>();
if(root == null || root.val > target)
return pathlist;
ArrayList<Integer> path = new ArrayList<Integer>();
FindPath(root,target,pathlist,path);
return pathlist;
}
public void FindPath(TreeNode root,int target,
ArrayList<ArrayList<Integer>> pathlist,ArrayList<Integer> path) {
if(root == null || root.val > target){
path.clear();
} else if(root.val == target) {
if(root.left == null && root.right == null) {
path.add(root.val);
pathlist.add(path);
} else {
path.clear();
}
} else {
path.add(root.val);
ArrayList<Integer> path2 = new ArrayList<Integer>();
path2.addAll(path);
target -= root.val;
FindPath(root.left,target,pathlist,path);
FindPath(root.right,target,pathlist,path2);
}
}
}