LeetCode-437-路径总和III

在这里插入图片描述

1、深度遍历

考虑到每条路径的节点不一定是从根节点开始的,因此我们需要遍历整棵二叉树分别把每个节点当作是根节点进行处理。我们在进行深度遍历时,利用rootSum遍历单个节点作为根节点时,节点值之和等于targetSum 的路径数目,为此我们在递归时采用temp_path记录路径数。最终我们遍历完所有可能并返回路径数。

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

    int pathSum(TreeNode *root, long targetSum) {
        int pathNum = 0;
        if (!root) return pathNum;
        pathNum += rootSum(root, targetSum);
        pathNum += pathSum(root->left, targetSum);
        pathNum += pathSum(root->right, targetSum);
        return pathNum;
    }
};

2、前缀和

我们在利用深度遍历搜索整棵二叉树时,很多的操作实际上是重复的。为了避免重复的操作,我们可以在以此深度遍历中,利用哈希表来维护我们遍历过路径的前缀和。我们可以利用哈希表来记录在进行深度遍历时,从根节点到某个节点的父节点中所有节点的值的和,对应这样前缀路径的数目。因此当我们在深度遍历整棵树时,我们只需要比较在当前节点的前缀和减去targetSum 后,在当前维护的前缀和哈希表中有多少条这样的路径即可。当我们遍历完这个节点之后的所有节点后,需要从哈希表中减去先前记录的路径从而保证这些前缀和都在同一条路径上。

class Solution {
public:
    unordered_map<long long, int> prefix;

    int dfs(TreeNode *root, long long curr, int targetSum) {
        if (!root) {
            return 0;
        }

        int ret = 0;
        curr += root->val;
        if (prefix.count(curr - targetSum)) {
            ret = prefix[curr - targetSum];
        }

        prefix[curr]++;
        ret += dfs(root->left, curr, targetSum);
        ret += dfs(root->right, curr, targetSum);
        prefix[curr]--;

        return ret;
    }

    int pathSum(TreeNode* root, int targetSum) {
        prefix[0] = 1;
        return dfs(root, 0, targetSum);
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值