BZOJ 1004 [HNOI2008]Cards

1004: [HNOI2008]Cards

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 1465   Solved: 858
[ Submit][ Status]

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种洗牌法中的一种代替，且对每种
洗牌法，都存在一种洗牌法使得能回到原状态。

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

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。

Source

[ Submit][ Status]

 

 

题意：有n个卡片要你为他们染上sr种红色，sb中蓝色，sg种绿色，（sr+sb+sg=n），然后给出一些等价的置换方式，问你最后又多少种染色方案。

 

思路：这道题相当于就是要我们算出等价类的数目，有多少个等价类，就是有多少个染色方案。等价类怎么算？用Burnside引理：等价类的数目=对应每一种置换的不动点数目/置换方案的数目   其中不动点的意思是经过这个置换后，染色的情况并没有发生变化。有了这个定理我们可以开始做了，怎么找出一种置换方案有多少个不动点呢？我们可以根据一个置换找出那些位置应该图一样的颜色，然后我们有三种颜色，问题就相当于是你有三个背包，每个背包有自己的容量（对应三种颜色的数量），然后又一些物品（一样颜色的位置组成一个物品），重量是这些相同颜色的位置的数目。我们要把物品装进背包里面，问刚刚好装满背包有多少方案。这里用动态规划就好了，最后的话，由于要模一个数，而且有除法，所以要找(m+1)（+1是因为自己要加一个置换方式:1->1,2->2,3->3....)在模p下的逆元.

 

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string.h>
#include<queue>
#include<algorithm>
#include<vector>
using namespace std;
#define LL long long
const int maxn = 20 + 2;
int dp[62][maxn][maxn][maxn];
int w[110], size;
int A, B, C, p, m, inv,n;

int qpow(int x, int q)
{
	int ret = 1, base = x;
	while (q > 0) {
		if (q & 1) ret = (ret*base) % p;
		base = (base*base)%p;
		q >>= 1;
	}
	return ret;
}

int dfs(int a, int b, int c, int i)
{
	if (a > A || b > B || c > C) return 0;
	if (a == A&&b == B&&c == C) return 1;
	if (i >= size) return 0;
	if (dp[i][a][b][c] != -1) return dp[i][a][b][c];
	dp[i][a][b][c] = (((dfs(a + w[i], b, c, i + 1) + dfs(a, b + w[i], c, i + 1))%p + dfs(a, b, c + w[i], i + 1))%p)%p;
	return dp[i][a][b][c];
}

vector<int> adj[maxn*3];
bool vis[maxn * 3];

bool dfs1(int u)
{
	if (vis[u]) return true;
	w[size]++;
	vis[u] = true;
	for (int i = 0; i < adj[u].size(); ++i) {
		int v = adj[u][i];
		if (dfs1(v)) return true;
	}
	w[size]--;
	vis[u] = false;
	return false;
}

void GetCircle()
{
	memset(vis, 0, sizeof(vis));
	size = 0;
	for (int i = 1; i <= n; ++i) if (!vis[i])
	{
		w[size] = 0;
		dfs1(i);
		++size;
	}
}

void input()
{
	size = n;
	memset(dp, -1, sizeof(dp));
	for (int i = 0; i < size; ++i) w[i] = 1;
	int ans = dfs(0, 0, 0, 0);
	for (int i = 0; i < m; ++i) {
		for (int j = 1; j <= n; ++j) {
			adj[j].clear();
			int x; scanf("%d", &x);
			adj[j].push_back(x);
		}
		GetCircle();
		memset(dp, -1, sizeof(dp));
		ans = (ans+dfs(0, 0, 0, 0))%p;
	}
	printf("%d\n", (ans * inv)%p);
}

int gcd(int a, int b)
{
	while (a&&b) {
		if (a > b)a %= b;
		else b %= a;
	}
	return a + b;
}

int main()
{
	while (scanf("%d%d%d%d%d", &A, &B, &C, &m, &p) == 5) {
		int cnt = 0;
		for (int i = 1; i < p;++i)
		if (gcd(i, p) == 1) ++cnt;
		inv = qpow((m+1), cnt - 1);
		n = A + B + C;
		input();
	}
}