Leetcode 523. 连续的子数组和 C++

Leetcode 523. 连续的子数组和

题目

给定一个包含 非负数 的数组和一个目标 整数 k，编写一个函数来判断该数组是否含有连续的子数组，其大小至少为 2，且总和为 k 的倍数，即总和为 n*k，其中 n 也是一个整数。

测试样例

示例 1：

输入：[23,2,4,6,7], k = 6
输出：True
解释：[2,4] 是一个大小为 2 的子数组，并且和为 6。

示例 2：

输入：[23,2,6,4,7], k = 6
输出：True
解释：[23,2,6,4,7]是大小为 5 的子数组，并且和为 42。

说明：

• 数组的长度不会超过 10,000 。
• 你可以认为所有数字总和在 32 位有符号整数范围内。

题解

前缀和
我们先获取前缀和，分两种情况进行讨论

• k==0时，我们只需要遍历数组，看是否存在连续两个及以上的0
• 其他情况下，我们判断区间[j,i]其和是否为k的正数倍，我们固定i，找j进行检查

详细过程见代码

代码

bool checkSubarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        if(n == 0)  return false;
        vector<int> preSum(n,0);
        preSum[0] = nums[0];
        for(int i=1; i<n; i++){
            preSum[i] = preSum[i-1]+nums[i];
        }
        int i=0,j;
        if(k == 0){			//是否存在连续两个及以上的0
            while(i < n){
                if(nums[i]!=0) i++;
                else{
                    for(j=i+1; j<n; j++){
                        if(nums[j] != 0)    break;
                    }
                    if(j-i > 1) return true;
                    i = j;
                }
            }
        }else{
            for(int i=1; i<n; i++){
                for(int j=0; j<i; j++){
                    if((preSum[i]-preSum[j]+nums[j])%k == 0)
                        return true;
                }
            }
        }
        return false;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/continuous-subarray-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。