2021-10-18 路径总和I+II+III

部分解题思路参考自LeetCode对应题目官方题解
LeetCode112-路径总和I
基础DFS,能优化的就是代码的简洁和美观了~

class Solution
{
public:
    bool hasPathSum(TreeNode *root, int targetSum)
    {
        if (!root)
            return false;
        if (!root->right && !root->left)
        {
            return targetSum == root->val;
        }
        return hasPathSum(root->left, targetSum - root->val) || hasPathSum(root->right, targetSum - root->val);
    }
};

LeetCode113-路径总和II
题目限定了路径的起始点为根节点和叶子节点,输出为路径,使用容器保存,并在深搜的同时实时更新。
总的思路就是DFS常规解法,有一个小点需要注意:当前为叶子节点时需检查当前路径是否符合要求,需要在更新容器、将当前节点保存并重新计算过路径和后进行,这样在当前节点所在的函数内就可以检查判断当前节点的路径。不然只能在其左右节点即为空节点进入函数时检查,这样会出现同一条路径重复出现的问题。
在这里插入图片描述

class Solution
{
public:
    vector<vector<int>> res;//储存结果
    vector<int> temp;//用于保存实时路径
    vector<vector<int>> pathSum(TreeNode *root, int targetSum)
    {
        dfs(root, targetSum, 0);
        return res;
    }
    void dfs(TreeNode *root, int targetSum, int cur)
    {
        if (!root)
            return;
        temp.push_back(root->val);
        cur += root->val;
        if (!root->left && !root->right)//当前为叶子节点时检查是否符合条件
        {
            if (cur == targetSum)//路径和恰好满足,放入结果容器
                res.push_back(temp);
        }
        dfs(root->left, targetSum, cur);
        dfs(root->right, targetSum, cur);
        temp.pop_back();//当前节点及其子树都已遍历过,退出前需更新temp容器保存的路径,删除当前节点
    }
};

LeetCode437-路径总和III
解法一:DFS
题目中所找路径可为途中任意自上而下连续的一段,也就是说任何一个节点都有可能作为路径的起点,对于任一起点,我们在内层使用DFS的方式找出满足的路径条数,同时外层使用DFS遍历整个树使得每一个节点都会作为起点进入内层的DFS搜寻!
时间复杂度:O(N^2)
空间复杂度:O(N)

class Solution {
public:
    int pathSum(TreeNode* root, int targetSum) {
        int res=0;
        if(root==nullptr) return 0;
        res+=rootSum(root,targetSum);
        res+=pathSum(root->left,targetSum);
        res+=pathSum(root->right,targetSum);
        return res;
    }
    int rootSum(TreeNode* root, int targetSum)
    {
        if(!root) return 0;
        int res=0;
        if(targetSum==root->val)
        {
            res=1;
        }
        res+=rootSum(root->left,targetSum-root->val);
        res+=rootSum(root->right,targetSum-root->val);
        return res;
    }
};

解法二:前缀和+DFS
在解法一中我们重复了许多路径和的计算,现在我们尝试优化时间消耗,去掉重复计算——保存前缀和
我们依旧使用DFS的思路遍历整个树,在便利的同时,我们保存当前遍历过并可连节路径的节点到根节点的路径和。对于遍历到的每个节点,我们搜寻保存下来的前缀和中,是否存在、存在几条满足:当前节点前缀和=某一结点前缀和+TargetSum 的节点。
注意的细节有:我们使用哈希表保存前缀和以及相应的个数,在退出每个节点时,需要及时更新保存的前缀和。
时间复杂度:O(N)
空间复杂度:O(N)

class Solution
{
public:
    map<long long, int> prefix;
    int Res(TreeNode *root, int curr, int targrtSum)
    {
        if (!root)
            return 0;
        int res = 0;
        curr += root->val;
        if (prefix.count(curr - targrtSum))
        {
            res += prefix[curr - targrtSum];
        }
        prefix[curr]++; //放在这里而不能放在if前是因为避免targetSum为零时出错!
        res += Res(root->left, curr, targrtSum);
        res += Res(root->right, curr, targrtSum);
        prefix[curr]--; //退出当前节点一定记得更新map删除相应的前缀和
        return res;
    }
    int pathSum(TreeNode *root, int targetSum)
    {
        prefix[0] = 1; //对应前缀和刚好等于targetSum时计次
        return Res(root, 0, targetSum);
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值