LeetCode 930 和相同的二元子数组[动态规划 前缀和] HERODING的LeetCode之路

在这里插入图片描述解题思路:
这题可以用多种方法去解决,第一种是利用哈希表实现的前缀和方法,前缀和最大的好处是可以通过前缀和之间的差是否为goal判断序列是否符合标准,即寻找当前前缀和sum - goal是否存在,代码如下:

class Solution {
public:
    int numSubarraysWithSum(vector<int>& nums, int goal) {
        int len = nums.size();
        vector<int> sum(len + 1);
        // 计算前缀和
        for(int i = 0; i < len; i ++) {
            sum[i + 1] = sum [i] + nums[i]; 
        }
        // 哈希表统计
        unordered_map<int, int> mp;
        int ans = 0;
        for(int& s : sum) {
            if(mp.count(s - goal)) {
                ans += mp[s - goal];
            }
            mp[s] ++;
        }
        return ans;
    }
};

这种方法甚至不需要数组存储前缀和的值,直接定义一个sum存储即可,代码如下:

class Solution {
public:
    int numSubarraysWithSum(vector<int>& nums, int goal) {
        int len = nums.size();
        int sum = 0;
        // 哈希表统计
        unordered_map<int, int> mp;
        mp[0] ++;
        int ans = 0;
        for(int i = 0; i < len; i ++) {
            sum += nums[i];
            if(mp.count(sum - goal)) {
                ans += mp[sum - goal];
            }
            mp[sum] ++;
        }
        return ans;
    }
};

动态规划的方法虽说是动态规划,本质还是前缀和形式,dp[i]代表前缀和为i的次数,代码如下:

class Solution {
public:
    int numSubarraysWithSum(vector<int>& nums, int goal) {
        int len = accumulate(nums.begin(), nums.end(), 0);
        vector<int> dp(len + 1, 0);
        dp[0] = 1;
        int ans = 0;
        int sum = 0;
        for(int& num : nums) {
            sum += num;
            if((sum - goal >= 0) && dp[sum - goal]) {
                ans += dp[sum - goal];
            }
            dp[sum] ++;
        }
        return ans;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

HERODING77

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值