K倍区间 -- 蓝桥杯

Posted 2022-04-01 满天星！

K倍区间 – 蓝桥杯

题目描述：
给定一个长度为 N 的数列，A1,A2,…AN，如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 K 的倍数，我们就称这个区间 [i,j] 是 K 倍区间。
你能求出数列中总共有多少个 K 倍区间吗？

输入格式
第一行包含两个整数 N 和 K。
以下 N 行每行包含一个整数 Ai。

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

数据范围
1≤N,K≤100000,
1≤Ai≤100000
输入样例：

5 2
1
2
3
4
5

输出样例：

6

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
ll s[N],cnt[N];
int main()

    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    
        cin>>s[i];
        s[i]+=s[i-1];
    
    ll ans=0;
    for(int i=0;i<=n;i++)
    
        ans+=cnt[s[i]%k];
        cnt[s[i]%k]++;
    
    cout<<ans<<endl;
    return 0;

如果觉得写的还不错，点个赞吧 ^ v ^