【BZOJ1004】Cards

polya计数+DP的入门题,话说网上几乎所有介绍polya计数的网页都无法直视。。。推荐自己去买一本组合数学的教材或者国家集训队的讲稿。。。

题目输入中“输入数据保证任意多次洗牌都可用这 m种洗牌法中的一种代替,且对每种洗牌法,都存在一种洗牌法使得能回到原状态。”等价于所有方案加上一个不变的置换就是一个置换群,这些重要条件为什么要这么含蓄。。。

其实洗牌的方式不用存下来的,录入一个DP一个也可以。。。第一次写没想这么多。。。懒得写二分快速幂了,写完暴力才发现长度差不多hhh



#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
int SA,SB,SC,m,p,n;
struct data
{
	int a[65];
	data()
	{
		memset(a,0,sizeof(a));
	}
}A[65];

void Init()
{
	scanf("%d%d%d%d%d",&SA,&SB,&SC,&m,&p);
	n=SA+SB+SC;
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n;j++)
		{
			scanf("%d",&A[i].a[j]);
		}
	}
	m++;
	for(int j=1;j<=n;j++)
	{
		A[m].a[j]=j;
	}
	return ;
}
int dp[25][25][25];
int qkpower(int a,int x,int mo)
{
	int ans=1;
	for(int i=1;i<=x;i++)
	{
		ans=(ans*a)%mo;
	}
	return ans;
}
void work()
{
	bool v[65];
	int cc[65],cnt,sum=0;
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n;j++)
		{
			v[j]=0; cc[j]=0;
		}
		cnt=0;
		int t;
		for(int j=1;j<=n;j++)if(!v[j])
		{
			cnt++; 
			t=A[i].a[j];
			while(t!=j)
			{
				cc[cnt]++; v[t]=1; t=A[i].a[t];
			}
			cc[cnt]++; v[t]=1; 
		}
		for(int u=0;u<=SA;u++)
		{
			for(int v=0;v<=SB;v++)
			{
				for(int w=0;w<=SC;w++)
				{
					dp[u][v][w]=0;
				}
			}
		}
		dp[0][0][0]=1;
		for(int k=1;k<=cnt;k++)
		{
			for(int v=SB;v>=0;v--)
			{
				for(int w=SC;w>=0;w--)
				{
					for(int u=SA;u>=0;u--)
					{
						if(cc[k]+u<=SA)
						{
							dp[u+cc[k]][v][w]=(dp[u+cc[k]][v][w]+dp[u][v][w])%p;
						}
						if(cc[k]+v<=SB)
						{
							dp[u][v+cc[k]][w]=(dp[u][v+cc[k]][w]+dp[u][v][w])%p;
						}
						if(cc[k]+w<=SC)
						{
							dp[u][v][w+cc[k]]=(dp[u][v][w+cc[k]]+dp[u][v][w])%p;
						}
					}
				}
			}
		}
		sum=(sum+dp[SA][SB][SC])%p;
	}
	sum=(sum*qkpower(m,p-2,p))%p;
	printf("%d\n",sum);
	return ;
}
int main()
{
	freopen("in.txt","r",stdin);
	Init();
	work();
	return 0;
}


你可能感兴趣的:(【BZOJ1004】Cards)