1. K倍区间

题目描述

给定一个长度为 $N$ 的数列，$A_1,A_2,\dots A_N$，如果其中一段连续的子序列$A_i,A_{i+1},\dots A_j$之和是 $K$ 的倍数，我们就称这个区间 $[i,j]$ 是 $K$ 倍区间。

你能求出数列中总共有多少个 $K$ 倍区间吗？

输入格式

第一行包含两个整数 $N$ 和 $K$ 。

以下 $N$ 行每行包含一个整数 $A_i$ 。

输出格式

输出一个整数，代表 $K$ 倍区间的数目。

数据范围

$1\le N,K\le 100000$
$1\le Ai\le 100000$

时/空限制

$1s/64MB$

输入样例

5 2
1
2
3
4
5

输出样例

6

---

解题思路：先暴力再优化

$1.$(暴力枚举) $O(n^3)$ （超时）

#include<iostream>

using namespace std;

int A[100010];

int main()
{
    int n,k; cin>>n>>k;
    for(int i=1;i<=n;i++)scanf("%d",&A[i]);
    
    long long ans=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=i;j++)
        {
            long long sum=0;
            for(int k=j;k<=i;k++) sum+=A[k];
            if(sum % k ==0) ans++;
        }
    
    cout<<ans;
    
    return 0;
}

$2.$(暴力枚举 + 前缀和优化) $O(n^2)$ （超时）

#include<iostream>

using namespace std;

int A[100010];
long long S[100010];

int main()
{
    int n,k; cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&A[i]);
        S[i]=S[i-1]+A[i];
    }
    
    long long ans=0;
    for(int i=1;i<=n;i++)
        for(int j=0;j<i;j++)
            if((S[i]-S[j])%k==0)
                ans++;
    
    cout<<ans;
    
    return 0;
}

$3.$(暴力枚举 + 前缀和优化 + 空间换时间优化) $O(n)$ （通过）

当$R$固定时，在$0$ ~ $R-1$之间找到有多少个$L$满足(s[R] - s[L]) % k == 0,这其实就是我们第二层循环的含义

可转换为：s[R] % k == s[L] % k

即有多少个$S[L]$与$S[R]$模k的余数相同

我们可以开一个新的数组$cnt[i]$，表示余数是i的数有多少个，从而实现以空间换时间的优化

#include<iostream>

using namespace std;

int A[100010];
long long S[100010];
int cnt[100000];

int main()
{
    int n,k; cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&A[i]);
        S[i]=S[i-1]+A[i];
    }
    
    cnt[0]++;
    long long ans=0;
    for(int i=1;i<=n;i++)
    {
        int yu=S[i]%k;
        ans+=cnt[yu];
        cnt[yu]++;
    }
    
    cout<<ans;
    
    return 0;
}