有一个环,环上有n个位置,每个位置有一个数,求有多少连续相邻的数组成的数是m的倍数,比如n=5 (9 6 4 2 8 ),m=7,有5个数(42, 28, 896, 42896 and 89642)。
比赛的时候一直不知道怎么处理位数超过总位数(n个数全部包含)的情况。
看了题解才明白,同过记录每一位的位数,然后求出总位数,每次去掉超过总位数的情况。
比如初始化s=64289,位数满足情况,下次更新6的余数的时候,减去642894的情况,通过s(64289)可以找到642896,即s=s*6位数+6.然后让位数为s的减1.每次都减去不符合的情况,到最后肯定都是合法的情况。。
dp[i][j],表示以a[i]为结尾的,余数为j的总数。。。
注意初始化的时候,不能把所有的dp即50001*200初始化为0,会超时,应该用memset(dp,0,sizeof(int)*n*200),写的时候二了,写成了*n*m。。。。0.0
#include"stdio.h"
#include"string.h"
#define N 50005
int dp[N][205];
int e[N*3];//10^i
int a[N];//
int l[N];//l[i]=a[i]的位数
int fun(int x)//求位数
{
if(x==0)return 1;
int t=0;
while(x)
{
x/=10;t++;
}
return t;
}
int main()
{
int i,j;
int n,m;
while(scanf("%d%d",&n,&m)!=-1)
{
memset(dp,0,sizeof(int)*n*205);
e[0]=1;
for(i=1;i<=n*3;i++)
e[i]=(e[i-1]*10)%m;
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
l[i]=fun(a[i]);
}
a[n]=a[0];
l[n]=l[0];
int s=0;
int ll=0;
int ans=0;
for(i=n;i;i--)
{
s=(a[i]*e[ll]+s)%m;//s=e[当前已有的长度]*a[i]+s, 123=1*100+23
ll+=l[i];
dp[0][s]++;
}
ans+=dp[0][0];
for(i=1;i<n;i++)
{
for(j=0;j<m;j++)
dp[i][(j*e[l[i]]+a[i])%m]+=dp[i-1][j];
s=(s*e[l[i]]+a[i])%m;//s=s*10^a[i]+a[i],即692896=69289*10^1+6;
dp[i][s]--;
dp[i][a[i]%m]++;//注意加上a只有a[i]的情况
s=((s-a[i]*e[ll])%m+m)%m;
ans+=dp[i][0];
}
printf("%d\n",ans);
}
return 0;
}