BZOJ1004: [HNOI2008]Cards

1004: [HNOI2008]Cards

Description

　　小春现在很清闲,面对书桌上的N张牌,他决定给每张染色,目前小春只有3种颜色:红色,蓝色,绿色.他询问Sun有
多少种染色方案,Sun很快就给出了答案.进一步,小春要求染出Sr张红色,Sb张蓝色,Sg张绝色.他又询问有多少种方
案,Sun想了一下,又给出了正确答案. 最后小春发明了M种不同的洗牌法,这里他又问Sun有多少种不同的染色方案.
两种染色方法相同当且仅当其中一种可以通过任意的洗牌法(即可以使用多种洗牌法,而每种方法可以使用多次)洗
成另一种.Sun发现这个问题有点难度,决定交给你,答案可能很大,只要求出答案除以P的余数(P为质数).

Input

　　第一行输入 5 个整数：Sr,Sb,Sg,m,p(m<=60,m+1<p<100)。n=Sr+Sb+Sg。
接下来 m 行，每行描述一种洗牌法，每行有 n 个用空格隔开的整数 X1X2...Xn，恰为 1 到 n 的一个排列，
表示使用这种洗牌法，第 i位变为原来的 Xi位的牌。输入数据保证任意多次洗牌都可用这 m种洗牌法中的一种代
替，且对每种洗牌法，都存在一种洗牌法使得能回到原状态。

Output

　　不同染法除以P的余数

Sample Input

1 1 1 2 7
2 3 1
3 1 2

Sample Output

2

HINT

　　有2 种本质上不同的染色法RGB 和RBG，使用洗牌法231 一次可得GBR 和BGR，使用洗牌法312 一次 可得BRG 

和GRB。

100%数据满足 Max{Sr,Sb,Sg}<=20。

Burnside引理

由于输入保证任意多次洗牌都可以由这m种洗牌法中的一种代替

所以加上自身置换 这m+1种置换可构成一个置换集合

考虑求每种置换的不定点个数

同一轮换间的元素颜色应相同

且同种颜色的点的个数满足限制

定义f[r][b][g]表示 颜色个数分别为r,b,g的方案数

01背包式转移即可

#include 
   
   
    
    
const int MaxN = 66;
int R, B, G, n, m, Mod, qr, que[MaxN], res;
int a[MaxN], Rota[MaxN], f[22][22][22];
int pow(int x, int k) {
	int res = 1, r = x;
	for (; k; k >>= 1, r = r * r % Mod)
		if (k & 1) res = res * r % Mod;
	return res;
}
int main()
{
	scanf("%d%d%d", &R, &B, &G);
	n = R + B + G;
	scanf("%d%d", &m, &Mod);
	for (int i = 1; i <= m; ++i) {
		for (int i = 1; i <= n; ++i)
			scanf("%d", &a[i]);
		
		Rota[0] = 0;
	    for (int i = 1; i <= n; ++i)
			if (a[i]) {
			    que[qr = 0] = i;
				for (int ql = 0; ql <= qr; ++ql) {
					int x = que[ql];
					if (a[x]) que[++qr] = a[x], a[x] = 0;
				}
				Rota[++Rota[0]] = qr;
			}
			
		for (int r = 0; r <= R; ++r)
			for (int b = 0; b <= B; ++b)
				for (int g = 0; g <= G; ++g)
					f[r][b][g] = 0;
		f[0][0][0] = 1;
		
		for (int i = 1; i <= Rota[0]; ++i) {
			for (int r = R; r >= 0; --r)
				for (int b = B; b >= 0; --b)
					for (int g = G; g >= 0; --g) {
						if (r >= Rota[i]) f[r][b][g] += f[r - Rota[i]][b][g];
						if (b >= Rota[i]) f[r][b][g] += f[r][b - Rota[i]][g];
						if (g >= Rota[i]) f[r][b][g] += f[r][b][g - Rota[i]];
						f[r][b][g] %= Mod;
					}
		}		
		res += f[R][B][G];
		res %= Mod;
		
	}
	
	for (int i = 1; i <= n; ++i) {
			for (int r = R; r >= 0; --r)
				for (int b = B; b >= 0; --b)
					for (int g = G; g >= 0; --g) {
						if (r) f[r][b][g] += f[r - 1][b][g];
						if (b) f[r][b][g] += f[r][b - 1][g];
						if (g) f[r][b][g] += f[r][b][g - 1];
					    f[r][b][g] %= Mod;
					}
		}		
	res += f[R][B][G];
	res %= Mod;
	
	res = res * pow(m + 1, Mod - 2) % Mod;
	printf("%d\n", res);
}