LeetCode - 523 连续的子数组和

Kreastine

于 2022-05-27 10:57:15 发布

阅读量109

点赞数

分类专栏： C语言文章标签： leetcode 算法哈希算法

本文链接：https://blog.csdn.net/weixin_42800817/article/details/124968466

版权

C语言专栏收录该内容

15 篇文章 0 订阅

订阅专栏

题目

在这里插入图片描述

思路

这题暴力解法是遍历所有的子数组以及计算其和，每次判断是否符合条件（是K 的倍数），但这种做法会超时，需要优化。

如果事先计算出数组 $n u m s$ 的前缀和数组，则对于任意一个子数组，都可以在 O(1) 的时间内得到其元素和。用 $p r e f i x S u m s [i]$ 表示数组 $n u m s$ 从下标 0 到下标 i 的前缀和，则 $n u m s$ 从下标 $p + 1$ 到下标 $q$ （其中 p<q）的子数组的长度为 q−p，该子数组的元素和为 $p r e f i x S u m s [q] - p r e f i x S u m s [p]$ 。

当 $p r e f i x S u m s [q] - p r e f i x S u m s [p]$ 为 k 的倍数时， $p r e f i x S u m s [p]$ 和 $p r e f i x S u m s [q]$ 除以 k 的余数相同。
因此只需要计算每个下标对应的前缀和除以 k 的余数即可，使用哈希表存储每个余数第一次出现的下标。

接下来遍历 $n u m s$ 的每个数，计算每个下标对应的前缀与 k 的余数：
①当前余数在哈希表里已经存在，说明此时找到了出现相同余数的两个前缀和，则可以计算两个前缀和之间的子数组的长度，长度 = 两个相同余数前缀和的下标之差。若这个长度大于2，则返回true；
②若余数不存在，则将当前余数和当前下标存入哈希表。

哈希表的设计方式：hash下标为余数，下标对应元素存储每个余数第一次出现的 $n u m s$ 的下标。

在构思的时候要考虑到三种情况：连续0，和为k倍数的子数组从nums 开头开始，和为k倍数的子数组是 nums 中间一小段。

实现代码

①

bool checkSubarraySum(int* nums, int numsSize, int k){
    int sums[numsSize+1];
    sums[0]=0;
    for (int i =0; i<numsSize; i++) {
        if(i>0)
        {   /*这里处理连续超过2个0的情况；*/
        if(nums[i]==0&&nums[i-1]==0 ){
              return true;
           }   
        }

        sums[i+1]=sums[i]+nums[i];
        if (sums[i+1]%k==0&&i>1) {/*这里处理了符合条件的子数组从nums头开始的情况*/
            return true;
        }
    }
    
    for (int i =0; i<=numsSize; i++) { 
        if(sums[i]<k){
            continue;/*这步优化很关键：省去第一次出现k倍数之前的前缀和的处理*/
        }
        for(int j=0;  j<i; j++) {
            /*这一步是题目条件：当子数组的和为k的倍数 且子数组长度超过1（大于等于2）*/
            if ((sums[i]-sums[j])%k ==0&&(i-j)>1) {
                return true;/*这处理了符合条件的子数组是nums中间的一段的情况*/
            }   
        }
    }
    return false;
}

continue相当于一直在找，若小于k，则不可能出现k倍数，直接跳出此次 for 循环，i 加一进入下一次；找到大于 k 的前缀和的时候，说明找到了包括 k 的倍数的前缀和（官方题解里说要在hash表里记录第一次出现K倍数的前缀和的下标，找到第二次k倍数前缀和时，求这两个前缀和的距离 - 下标相减，这里相当于找到了第二次出现k倍数的前缀和，所以从i开始往前遍历查找，找第一次出现k倍数的前缀和的下标，看有没有长度符合条件的）。