蓝桥杯试题 历届试题 k倍区间(第八届蓝桥杯省赛C++B组)

k倍区间

题目链接-k倍区间
资源限制
时间限制：2.0s 内存限制：256.0MB
问题描述
　　给定一个长度为N的数列，A1, A2, … AN，如果其中一段连续的子序列Ai, Ai+1, … Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。

你能求出数列中总共有多少个K倍区间吗？
输入格式
　　第一行包含两个整数N和K。(1 <= N, K <= 100000)
　　以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)
输出格式
　　输出一个整数，代表K倍区间的数目。
样例输入

5 2
1
2
3
4
5

样例输出

6

解题思路

前缀和

• 我们可以用pre[]数组来记录前缀和，即pre[i]为前 i 项和, pre[R]−pre[L−1]就是区间[L,R]的和
• (pre[R]-pre[L-1])%k==0 说明这个区间是K倍区间，即pre[R]%k == pre[L-1]%k
• 所以我们能得到对前缀和取模之后，两个相等的前缀和就能组成一个k倍区间
• 开一个计数数组cnt[]，cnt[i]就是计算余数为i的前缀和数目，
• cnt[0]表示的是从第一个到当前的和就是可以整除k的区间数目，所以最后要加上
• 具体操作见代码

附上代码

#include<bits/stdc++.h>
#define int long long
#define lowbit(x) (x &(-x))
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=1e5+5;
typedef long long ll;
typedef pair<int,int> PII;
int a[N],pre[N],cnt[N];
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	int n,k,ans=0;
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		pre[i]=(pre[i-1]+a[i])%k;
		ans+=cnt[pre[i]];//加上在此之前与它同余的前缀和
		cnt[pre[i]]++;//统计前缀和模k后的余数出现次数 
	}
	cout<<ans+cnt[0]<<endl;
	return 0;
}