题目:http://acm.hdu.edu.cn/showproblem.php?pid=4669
题目大意:给你n个数字,顺时针的,即n个数字成环,然后在给你一个k,问你在这个环中选取一段数字,拼起来时mod k ==0 的方案数。
思路:很明显是DP,如果不考虑成环,那么也很容易想到状态量及状态转移方程,设d[ i ][ j ] 表示以第i个数字为结尾,余数为k的方案数,状态转移方程为 d[ i ][ ( num[ i ]*len[ i ] + j )%k ] =SIGMA( d[ i - 1][ j ] ),num[ i ] 表示第i个数,len[ i ]表示这个数有几个位数,也就是10的num[ i ]的长度次方。关键是成环这里不好处理,赶脚也没有什么想法来处理。之后,看了别人的博客才有所领悟。
先一遍按照上述DP处理过去,那么就算出了以第n个数为结尾的各种余数的方案数,然后再把这n个数字接上去那么再一遍处理过去,就相当于是成环了。可是光考虑到这一点还不够,因为一个这段数的数字个数不能超过n。像样例,接上去是:9 6 4 2 8 9 6 4 2 8,那么如果不考虑其他的,那么会跑出来6(亲身体会。。= =),会把4 2 8 9 6 4 2 这种情况加上去。也就是说还要减掉长度超过n的。就是这里,想了好久,也没有想到好的处理方法。。
无奈之下,就去看了看别人怎么处理的,发现真该学习一下:按照样例来说,第一遍是处理到8,第二遍处理,第一个数字9加进去以后,它跟据8来转移状态的,那么就应该把9 6 4 2 8 9 这种情况减掉,就是d[i][sum-num[i]*num_ten %k] -- ,这里的i>n,num[i-n]==num[i],sum是前面n个数拼起来的%k的值,接下来是6,那么就有两种非法转移了,一个是9 6 4 2 8 9 6,另一个是6 4 2 8 9 6,但是仔细想想,9 6 4 2 8 9 6 是不会出现的,因为前面9这个状态值已经改变,6是由9转移过来的,也就是只要处理6又当头又当尾的这种情况,这样处理方式就和前面相同了。那么,可以想到,之后的处理也是一样。
真的学习了,因为自己就是长度大于n这里不会处理,如果标记位置的话(我的第一想法。。),后面会越来越多的,肯定不行。。
太帅了,mark一下,嘿嘿。。
代码如下:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 55555 ;
int len[MAXN];
int d[2][222];
int num[MAXN];
int main()
{
int n,k;
while(~scanf("%d%d",&n,&k))
{
memset(len,0,sizeof(len));
int num_ten = 1;
for(int i = 1;i<=n;i++)
{
scanf("%d",&num[i]);
int tmp = num[i];
len[i] = 1;
while(tmp)
{
len[i] = len[i]*10;
num_ten = num_ten*10%k;
tmp /= 10;
}
}
int cur = 0,pre = 0;
memset(d[pre],0,sizeof(d[pre]));
for(int i = 0;i<n;i++)
{
cur ^= 1;
memset(d[cur],0,sizeof(d[cur]));
d[pre][0] ++ ;
for(int j = 0;j<k;j++)
if(d[pre][j])
{
int to = (j*len[i+1]%k + num[i+1]%k)%k;
d[cur][to] += d[pre][j];
}
pre = cur;
}
int sum = 0;
for(int i = 0;i<n;i++)
sum = (sum*len[i+1]%k + num[i+1]%k)%k;
int ans = 0;
for(int i = 0;i<n;i++)
{
cur ^= 1;
memset(d[cur],0,sizeof(d[cur]));
d[pre][0] ++ ;
for(int j = 0;j<k;j++)
if(d[pre][j])
{
int to = (j*len[i+1]%k + num[i+1]%k)%k;
d[cur][to] += d[pre][j];
}
sum = (sum*len[i+1]%k + num[i+1]%k)%k;
d[cur][sum] -- ;
sum = (sum - num_ten*num[i+1]%k + k)%k ;
ans += d[cur][0];
pre = cur;
}
printf("%d\n",ans);
}
return 0;
}