題面

除了不洗牌以外，每種洗牌方式的每個循環裏的顏色必須一樣，然後大力背包一下就好了。最後記得把不洗牌的方案也算進去

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 using namespace std;
 5 const int N=64;
 6 int n,m,p,c1,c2,c3,ans;
 7 int dp[2][N][N][N],noww,last;
 8 int trs[N][N],vec[N][N],siz[N],vis[N];
 9 void exGCD(int
 
 a,int b,int &x,int &y)
10 {
11     if(!b) x=1,y=0;
12     else exGCD(b,a%b,y,x),y-=a/b*x;
13 }
14 int Inv(int x)
15 {
16     int xx,yy;
17     exGCD(x,p,xx,yy);
18     return (xx%p+p)%p;
19 }
20 void Mod(int &x,int y)
21 {
22     x+=y;
23     if(x>=p) x-=p;
24 }
25 void Getcir(int idx,int
 
 pos)
26 {
27     vis[pos]=true,vec[idx][siz[idx]]++;
28     if(!vis[trs[idx][pos]])
29         Getcir(idx,trs[idx][pos]);
30 }
31 int main()
32 {
33     scanf("%d%d%d%d%d",&c1,&c2,&c3,&m,&p),n=c1+c2+c3;
34     for(int i=1;i<=m;i++)
35     {
36         for(int j=1;j<=n;j++)
 
37             scanf("%d",&trs[i][j]);
38         memset(vis,0,sizeof vis);
39         for(int j=1;j<=n;j++)
40             if(!vis[j]) siz[i]++,Getcir(i,j);
41     }
42     for(int i=1;i<=m;i++)
43     {
44         memset(dp,0,sizeof dp),noww=dp[0][0][0][0]=1,last=0;
45         for(int j=1;j<=siz[i];j++)
46         {
47             int sz=vec[i][j];
48             for(int k=0;k<=c1-sz;k++)
49                 for(int h=0;h<=c2-sz;h++)
50                     for(int l=0;l<=c3-sz;l++)
51                     {
52                         int t=dp[last][k][h][l];
53                         if(k+sz<=c1) Mod(dp[noww][k+sz][h][l],t);
54                         if(h+sz<=c2) Mod(dp[noww][k][h+sz][l],t);
55                         if(l+sz<=c3) Mod(dp[noww][k][h][l+sz],t);
56                     }
57             last=noww,noww^=1;
58         }
59         ans+=dp[last][c1][c2][c3],ans%=p;
60     }
61     memset(dp,0,sizeof dp),noww=dp[0][0][0][0]=1,last=0;
62     for(int i=1;i<=n;i++)
63     {
64         for(int k=0;k<=c1;k++)
65             for(int h=0;h<=c2;h++)
66                 for(int l=0;l<=c3;l++)
67                 {
68                     int t=dp[last][k][h][l];
69                     if(k+1<=c1) Mod(dp[noww][k+1][h][l],t);
70                     if(h+1<=c2) Mod(dp[noww][k][h+1][l],t);
71                     if(l+1<=c3) Mod(dp[noww][k][h][l+1],t);
72                 }
73         last=noww,noww^=1;
74     }
75     printf("%lld",1ll*(ans+dp[last][c1][c2][c3])*Inv(m+1)%p);
76     return 0;
77 }

解題：HNOI 2013 Cards