剑指offer java版 test24—二叉树搜索路径
题目:
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路:
从根结点root开始,如果root的值比target小,则开始找root的左右子树,直到各值加起来刚好为target,且刚好是叶结点。如果走到叶结点还是不等于target,那么依次返回 再从根结点开始找
代码:
package cn.test24;
import java.util.ArrayList;
/*
输入一颗二叉树的根节点和一个整数,
打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(
注意: 在返回值的list中,数组长度大的数组靠前)
*/
public class test {
public static void main(String[] args){
//构建一个二叉树
Solution sl=new Solution();
TreeNode a=new TreeNode(5);
TreeNode b=new TreeNode(1);
TreeNode c=new TreeNode(10);
TreeNode d=new TreeNode(9);
TreeNode e=new TreeNode(6);
TreeNode f=new TreeNode(3);
TreeNode g=new TreeNode(2);
TreeNode h=new TreeNode(4);
a.left=b;
a.right=c;
b.left=d;
b.right=e;
e.right=f;
c.left=g;
c.right=h;
sl.FindPath(a,15);
}
}
class Solution {
ArrayList<ArrayList<Integer>> pathAll=new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> pathSingle=new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root==null){
return pathAll;
}
pathSingle.add(root.val);
target=target-root.val;
if(target==0&&root.left==null&&root.right==null){//如果到达叶结点,且各值加起来等于target 那么保存这条路径
pathAll.add(new ArrayList<Integer>(pathSingle));
}
if(root.left!=null) FindPath(root.left,target);//找左子树
if(root.right!=null) FindPath(root.right,target);//找右子树
pathSingle.remove(pathSingle.size()-1);//如果找到叶结点时,路径不满足,则退一步,即退回到根结点,再找别的路
return pathAll;
}
}
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}