【剑指offer】34.二叉树中和为某一值的路径

34.二叉树中和为某一值的路径

面试题34. 二叉树中和为某一值的路径

难度中等27

输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

示例:

给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

//回溯
// 递归解决 终止条件节点为null 返回
// 当target=0 并且左右子节点都为null的时候 进行数据添加。
// 否则的话 直接递归左右子节点。
List<List> ret = new ArrayList<>();

public List<List<Integer>> pathSum(TreeNode root, int sum) {
    pathSum(root,sum,new ArrayList());
    return ret;
}

public void pathSum(TreeNode root,int target,List<Integer> path){
    if(root == null){
        return;
    }
    path.add(root.val);
    target -= root.val;
    if(target == 0 && root.left == null && root.right == null){
        ret.add(new ArrayList<>(path));
    }else{
        pathSum(root.left,target,path);
        pathSum(root.right,target,path);
    }
    //回溯,状态重置,只添加不删除就是错的路径了,因为整个递归过程用的是同一个收集器path
    path.remove(path.size()-1);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章