2021-5-6 剑指 Offer 34. 二叉树中和为某一值的路径(回溯算法)

注:

        if(root->left==nullptr&&root->right==nullptr&&count==target){
            result.push_back(nums);
            //这里不可以加return,不然会导致少进行一个nums.pop_back()
        }

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

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

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

返回:

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

提示:
节点总数 <= 10000

题解:
从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

算法概述:
利用树的递归性质, 使用先序遍历, 将遍历经过的节点的 value 记录在数组 nums 中, 当到达叶子接节点后, if(sum(nums) == target) 则将 nums 记录在一个全局变量中 result, 遍历完整棵树后, result 即为解

递归的过程
首先使用先序遍历, 先序遍历是第一次访问到相应节点, 就进行操作的一种遍历方式, 回溯算法的题目基本都是采用先序遍历的方式。
从 root 节点开始, 当访问到一个节点时, 将节点的 value 记录在 nums 中, 如下图, 红色 的数组即为将相应节点的 value 记录到 nums 之后 nums 中的元素。
在这里插入图片描述

因为要记录所有经过的节点, 所以 nums 是要在递归过程中逐层的转递的, 所以当访问完某一个节点后,要将该节点的 value 从 nums 中删除。 如下图, 绿色 的数组, 即为将 value 从 nums 中删除的结果。
nums 中的元素, 通过上面两个步骤的操作, nums 中的元素始终都是代表从跟,到当前节点路径的所有元素
当到达任意叶子节点, 则计算 sum(nums) == target 如果相等, 则放入全局的 result 数组中

复杂度分析:
时间复杂度 O(N) : NN 为二叉树的节点数,先序遍历需要遍历所有节点。
空间复杂度 O(N) : 最差情况下,即树退化为链表时,path 存储所有树节点,使用 O(N)额外空间。

class Solution {
public:
    vector<vector<int>> result;
    vector<int> nums;
    void dfs(TreeNode* root,int target,int count){
        if(root==nullptr){
            return ;
        }        
        nums.push_back(root->val);
        count=count+root->val;
        if(root->left==nullptr&&root->right==nullptr&&count==target){
            result.push_back(nums);
            //这里不可以加return,不然会导致少进行一个nums.pop_back()
        }
        dfs(root->left,target,count);
        dfs(root->right,target,count);
        nums.pop_back();
    }
    vector<vector<int>> pathSum(TreeNode* root, int target) {
        if(root==nullptr){
            return {};
        }
        int count=0;
        dfs(root,target,count);
        return result;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值