剑指Offer算法题 26:二叉树中和为某一值的路径
题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
解题思路
方法一
递归。由于路径从根结点开始,也即每次得到的路径都是从根节点出发直至某个叶结点,所以应采取先序遍历的方式进行二叉树的遍历。将满足条件的路径添加到结果集result中最终返回即可。
时间复杂度:O(n);空间复杂度:O(n)
源码
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 {
private ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
private ArrayList<Integer> list = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root == null)return result;
list.add(root.val);
target -= root.val;
//叶子节点
if(target == 0 && root.left == null && root.right == null)
result.add(new ArrayList<Integer>(list));
//因为在每一次的递归中,我们使用的是相同的result引用,所以其实左右子树递归得到的结果我们不需要关心,
//可以简写为FindPath(root.left, target);FindPath(root.right, target);
FindPath(root.left, target);
FindPath(root.right, target);
list.remove(list.size()-1);
return result;
}
}
运行结果
原文始发于微信公众号(AJCoder):剑指Offer算法题 26:二叉树中和为某一值的路径