标题: k倍区间
给定一个长度为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
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 2000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。
注意:
main函数需要返回0;
只使用ANSI C/ANSI C++ 标准;
不要调用依赖于编译环境或操作系统的特殊函数。
所有依赖的函数必须明确地在源文件中
不能通过工程设置而省略常用头文件。
提交程序时,注意选择所期望的语言类型和编译器类型。
求出前缀和然后取模,范围0~k-1之间。
用ct数组记录之前i出现过几次。
pos数组记录i是否出现过,且第一次出现的位置
抽屉原理就是如果sum[i]=sum[j],那么[i+1,j]这一段必定能被k整除。
样例:
1
2
3
4
5 化成 1 1 0 0 1,例如sum[2]到sum[5],1,0,0,1,由于sum[2]==sum[5],那么(pos[sum[2]]+1,pos[sum[5]])这一段连续区间和被k整除,具体可以看抽屉原理。
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<ctime>
#include<string>
#include<stack>
#include<deque>
#include<queue>
#include<list>
#include<set>
#include<map>
#include<cstdio>
#include<limits.h>
#define MOD 1000000007
#define fir first
#define sec second
#define fin freopen("/home/ostreambaba/文档/input.txt", "r", stdin)
#define fout freopen("/home/ostreambaba/文档/output.txt", "w", stdout)
#define mes(x, m) memset(x, m, sizeof(x))
#define Pii pair<int, int>
#define Pll pair<ll, ll>
#define INF 1e9+7
#define inf 0x3f3f3f3f
#define Pi 4.0*atan(1.0)
#define lowbit(x) (x&(-x))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
typedef unsigned long long ull;
const double eps = 1e-9;
const int maxn = 1e7;
using namespace std;
inline int read(){
int x=0,f=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return x*f;
}
int sum[maxn],kt[maxn],pos[maxn],ct[maxn];
int main(){
int n,k;
sum[0]=0;
int ans=0;
fill(pos,pos+n+1,-1);
n=read(),k=read();
for(int i=1;i<=n;++i){
int t=read();
sum[i]=(sum[i-1]+t)%k;
}
for(int i=1;i<=n;++i){
if(sum[i]==0){
++ans;
pos[sum[i]]=i;
}if(pos[sum[i]]==-1){
pos[sum[i]]=i;
ct[sum[i]]++;
}else{
ans+=ct[sum[i]];
ct[sum[i]]++;
}
}
printf("%d\n",ans);
return 0;
}