二叉树中和为某一值的路径
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路:
当用前序遍历的方式访问某一节点的时候,我们该把节点添加到路径上,并累加该节点的值。
如果当前节点是叶子节点,并且路径中节点值的和刚好等于输入的整数,则当前路径符合要求,将其添加到数组中;
如果当前节点不是叶子节点,则继续访问它的子节点。当前节点访问结束之后,递归函数应该自动回到他的父节点。在退出函数之前要在路径上删除当前节点,并且减去当前节点的值,以确保返回父节点的额路径是从根节点到父节点。
package swordOffer;
import java.util.ArrayList;
/*
* 输入一颗二叉树的跟节点和一个整数,
* 打印出二叉树中结点值的和为输入整数的所有路径。
* 路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
* (注意: 在返回值的list中,数组长度大的数组靠前)
*
* */
public class FindPath {
static ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>();
//之前这个有问题是因为自己没有加static关键字
static ArrayList<Integer> list = new ArrayList<>();
@Override
public String toString() {
return "FindPath [getClass()=" + getClass() + ", hashCode()=" + hashCode() + ", toString()=" + super.toString()
+ "]";
}
public static 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));
}
//如果不是叶子结点,就遍历它的左右节点
FindPath(root.left, target);
FindPath(root.right, target);
list.remove(list.size()-1);//返回父节点前,在路径上删除当前节点
return listAll;
}
public static void main(String[] args) {
TreeNode tn = new TreeNode(10);
tn.left = new TreeNode(5);
tn.left.left = new TreeNode(4);
tn.left.right = new TreeNode(7);
tn.right = new TreeNode(12);
ArrayList<ArrayList<Integer>> al = FindPath(tn, 22);
System.out.println(al);
for(ArrayList<Integer> ll : al) {
System.out.println(ll);
}
}
}
博客围绕二叉树中和为某一值的路径问题展开。输入二叉树根节点和整数,需找出节点值和为该整数的所有路径。解题采用前序遍历,将节点添加到路径并累加值,若为叶子节点且和符合要求则记录路径,访问结束后从路径删除当前节点并减值。
370

被折叠的 条评论
为什么被折叠?



