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

题目

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

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

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

返回:

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

分析

注意的地方

1.如果是在节点为null时判断和是否为sum,那么一条路径会对应相同的两个输出,因为尾节点的左孩子和右孩子都是null。

2.保存路径的变量res是引用类型,在加入到ans中时应当new一个res的拷贝。否则后续对res的修改会改变ans里的内容。

3.res是共享的变量且为引用类型,从一条路径搜索完回到当前位置时它的值已经改变,因此需要回溯到原状态。

4.res回溯到原状态的方法:当前节点对res的修改是添加了cur.val,因此在遍历完以当前节点为根的子树后,需要删去cur.val,这样回溯到当前节点的父节点时res才没有改变。因为遍历一棵子树前和遍历完一棵子树后res状态不变,所以当前节点往res里添加的cur.val在遍历完左右子树后仍是末尾元素。

代码

class Solution {
    private List<List<Integer>> ans=new ArrayList<>();
    //cur为当前节点,s为根节点到当前节点元素值之和
    public void dfs(TreeNode cur,int s,int sum,ArrayList<Integer> res){
        if(cur==null) return;
        s+=cur.val;
        res.add(cur.val);
        if(s==sum&&cur.left==null&&cur.right==null) ans.add(new ArrayList<Integer>(res));
        dfs(cur.left,s,sum,res);
        dfs(cur.right,s,sum,res);
        //共享变量res被改变了,需要回溯到之前状态
        res.remove(res.size()-1);
    }
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        dfs(root,0,sum,new ArrayList<Integer>());
        return ans;
    }
}

LeetCode链接:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof

posted @ 2021-01-09 21:29  归鸿唱晚  阅读(70)  评论(0编辑  收藏  举报