BZOJ 1004 Cards(Polya定理)

题目

只要等价类中的所有元素都合法或者都不合法，那么Polya定理仍然可以统计合法的等价类数，直接DP求合法方案的不动点平均值。

AC Code:

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#define maxn 105
using namespace std;

int r,b,g,m,mod,ans,n;
int f[maxn][maxn][maxn],fac[maxn]={1,1},inv[maxn]={1,1},invf[maxn]={1,1},a[maxn],vis[maxn];

int main(){
	scanf("%d%d%d%d%d",&r,&b,&g,&m,&mod);
	n = r + b + g;
	for(int i=2;i<=n;i++)	
		fac[i] = 1ll * fac[i-1] * i % mod,
		inv[i] = 1ll * (mod - mod / i) * inv[mod % i] % mod,
		invf[i] = 1ll * invf[i-1] * inv[i] % mod;
	ans = fac[n] * 1ll * invf[r] % mod * invf[b] % mod * invf[g] % mod;
	for(int i=1;i<=m;i++){
		for(int j=1;j<=n;j++)
			scanf("%d",&a[j]);
		memset(f,0,sizeof f);
		f[0][0][0] = 1;
		for(int j=1;j<=n;j++)
			if(vis[j]!=i){
				int cnt = 1;
				vis[j] = i;
				for(int k=a[j];vis[k]!=i;k=a[k])
					vis[k] = i , cnt++;
				for(int a=r;a>=0;a--)
					for(int B=b;B>=0;B--)
						for(int c=g;c>=0;c--){
							if(a>=cnt) f[a][b][c] = (f[a][b][c] + f[a-cnt][b][c]) % mod;
							if(B>=cnt) f[a][B][c] = (f[a][B][c] + f[a][B-cnt][c]) % mod;
							if(c>=cnt) f[a][b][c] = (f[a][b][c] + f[a][b][c-cnt]) % mod;
						}
			}
		ans = (ans + f[r][b][g]) % mod;
	}
	ans = 1ll * ans * inv[m+1] % mod;
	printf("%d\n",(ans+mod)%mod);
}