LightOJ1125 Divisible Group Sums(DP)

题目问从N个数中取出M个数,有多少种取法使它们的和能被D整除。

dp[i][j][k]表示,前i个数取出j个数模D的余数为k的方案数

我用“我为人人”的方式来转移,就从i到i+1转移,对于第i+1个数有取和不取两种选择,然后确定j和k这两个维度的情况。

另外题目说数字是32位有符号整数,所以是会出现负数的。。。模D之后加D再模D就行了。

 1 #include<cstdio>
 2 #include<cstring>
 3 using namespace std;
 4 long long dp[200][11][20];
 5 int main(){
 6     int t,n,q,a[200],d,m;
 7     scanf("%d",&t);
 8     for(int cse=1; cse<=t; ++cse){
 9         scanf("%d%d",&n,&q);
10         for(int i=0; i<n; ++i) scanf("%d",a+i);
11         printf("Case %d:\n",cse);
12         while(q--){
13             scanf("%d%d",&d,&m);
14             memset(dp,0,sizeof(dp));
15             dp[0][1][(a[0]%d+d)%d]=1;
16             dp[0][0][0]=1;
17             for(int i=0; i<n-1; ++i){
18                 for(int j=0; j<=m; ++j){
19                     for(int k=0; k<d; ++k){
20                         if(j<m) dp[i+1][j+1][((k+a[i+1])%d+d)%d]+=dp[i][j][k];
21                         dp[i+1][j][k]+=dp[i][j][k];
22                     }
23                 }
24             }
25             printf("%lld\n",dp[n-1][m][0]);
26         }
27     }
28     return 0;
29 }

 

posted @ 2016-01-20 18:09  WABoss  阅读(162)  评论(0编辑  收藏  举报