LeetCode523. 连续的子数组和

因为子数组是连续的,所以判断连续的子数组的和时,我们往往开一个前缀和数组预处理出所有数的前缀和,这样能够降低求子数组的和的时间复杂度。

这题需要单独处理k为0的情况,由于数组所有元素都是非负数,所以当k为0时,如果存在两个相邻的数的值都为0,则返回true,否则返回false。

使用前缀和判断是否存在满足条件的子数组时,只需从小到大枚举子数组的长度,然后再在内层循环内枚举子数组的长度,然后使用前缀和计算区间和的公式计算是否存在子数组的和满足条件即可。有点类似区间dp,hh。

代码如下:

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        if(k == 0) {
            for(int i = 0; i < n; ++i) {
                if(nums[i] == 0 && i + 1 < n && nums[i + 1] == 0) {
                    return true;
                }
            }
            return false;
        }
        vector<int> preSum(n + 1, 0);                  // preSum是前缀和, preSum[i]表示nums[0] + nums[1] + ... + nums[i - 1]
        for(int i = 1; i <= n; ++i) {
            preSum[i] = preSum[i - 1] + nums[i - 1];
        }
        for(int len = 2; len <= n; ++len) {             // 从小到大枚举子数组的长度,注意,题目要求子数组的长度至少是2
            for(int i = 1; i + len - 1 <= n; ++i) {      // 枚举子数组的起点,i + len - 1是子数组的终点
                int j = i + len - 1;
                if((preSum[j] - preSum[i - 1]) % k == 0) {
                    return true;
                }
            }
        }
        return false;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值