1230. K倍区间★★★★

1230. K倍区间

写在前面: y总说蓝桥杯的题代码都不是很长,但是思路比较难想.需要花大量的时间放在思路上.一开始很难想出最优的解决方案,但是不要放弃.

cnt数组需要好好理解

难点: 如何优化掉第二重循环

for(int r=1;r<=n;r++)
    for(int l=1;l<=r;l++)
    {
        if(sum[r] - sum[l-1] % k ==0)
            res ++;
    }

第二重循环的目的是找1~r中有多少个l 满足sum[r] - sum[l-1] % k =0,也就是sum[r] % k = sum[l-1] % k.就是找1 ~ r中有多少个前缀和 % k等于一个确定的数 sum[r] % k.

方法: 用cnt[i]表示1 ~ i中有多少个前缀和(前缀和数组sum[]的前i个数)满足sum[j] % k == i( j 从1 ~ i).所以就用cnt数组代替了第二重循环. 所以顺序遍历sum数组,让res += cnt[sum[i] % k],然后让当前遍历到的cnt[sum[i] % k] ++
每遍历找到一个 sum[i] %k,就设置cnt[sum[i]%k]++,这也是用空间换时间的典型例子.然后后面在需要用的时候,就可以直接res += cnt[sum[i]%k]

假设从A₁,A₂…A_i的和正好是k倍区间,拿sum[i] % k==0,所以是cnt[0]是一个k倍区间.如果cnt[0]是0,那就忽略了1~i这个k倍区间.所以应该初始化cnt[0] = 1

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 100010;
ll sum[N];
ll cnt[N];
int n,k;
int main()
{

    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&sum[i]);
        sum[i] += sum[i-1];
    }
    
    ll ans = 0;    
    cnt[0] = 1;
    for(int i=1;i<=n;i++)
    {
        ans += cnt[sum[i] % k];
        cnt[sum[i] % k] ++;
    }
    printf("%lld",ans);
}