Description
给定一个数列,求是否存在连续子列和为m的倍数,存在输出YES,否则输出NO
Input
输入文件的第一行有一个正整数T(1≤T≤10),表示数据组数
接下去有T组数据,每组数据的第一行有两个正整数n,m ( 1≤n≤100000 , 1≤m≤5000)
第二行有n个正整数x (1≤x≤100)表示这个数列
Output
输出T行,每行一个YES或NO
Sample Input
2
3 3
1 2 3
5 7
6 6 6 6 6
Sample Output
YES
NO
Solution
开一个标记数组表示前缀和模m后的值,如果某个前缀和已经被标记过,说明前面有个不用的前缀和模m后的值与当前前缀和模m后的值相同,即说明存在一个子段和是m的倍数,注意初始化flag[0]=1
Code
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define maxn 111111
int T,n,m,a,sum,mark,flag[maxn];
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
sum=mark=0;
memset(flag,0,sizeof(flag));
flag[0]=1;
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
sum=(sum+a)%m;
if(!flag[sum])flag[sum]=1;
else mark=1;
}
printf("%s\n",mark?"YES":"NO");
}
return 0;
}