[ZJOI2019]麻将

这是一道麻将自动机的模板题(雾

其实这是一道dp套dp借助自动机实现的麻将好题!

首先把期望转化一下,拆成sigema p(x>i)

现在要计算i张牌不胡的概率,也就等价于计算i张牌不胡的方案数。

如果我们能建立一个关于麻将的自动机,支持插入麻将,判断当前牌型是否能胡,既可以在麻将自动机上dp解决本题。

先考虑如何压缩一副麻将,显然只需要把它的每一种牌型的个数算出来即可。

再考虑如何判定一副麻将是否能胡,这个可以借助dp。

dp[i][j][k]表示当前有i个x,还额外预留了j个(x,x-1),k表示是否有对子。

转移的时候枚举下一个位置预留几组,走个顺子即可。

现在我们要对这个过程建立自动机。

考虑两个麻将状态不同,当且仅当它们存在某种相同的状态,dp值却互不相同。

写一个结构体,然后bfs寻找状态+map判重即可。

不同的状态数不多,2000多种的样子。

建好这个东西后再去dp就可以了。

#include<bits/stdc++.h>
#define N 1100
#define M 5000
#define db double
#define ll long long
#define ldb long double
using namespace std;
inline int read()
{
	char ch=0;
	int x=0,flag=1;
	while(!isdigit(ch)){ch=getchar();if(ch=='-')flag=-1;}
	while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
	return x*flag;
}
const int mo=998244353,inf=1e9+7;
struct node
{
	int cnt,f[3][3][2];
	void clear()
	{
		cnt=0;
		for(int i=0;i<3;i++)for(int j=0;j<3;j++)for(int k=0;k<2;k++)f[i][j][k]=-inf;
	}
};
bool operator<(node a,node b)
{
	for(int i=0;i<3;i++)for(int j=0;j<3;j++)for(int k=0;k<2;k++)
	if(a.f[i][j][k]!=b.f[i][j][k])return a.f[i][j][k]<b.f[i][j][k];
	return a.cnt<b.cnt;
}
bool operator==(node a,node b)
{
	for(int i=0;i<3;i++)for(int j=0;j<3;j++)for(int k=0;k<2;k++)
	if(a.f[i][j][k]!=b.f[i][j][k])return false;
	if(a.cnt!=b.cnt)return false;
	return true;
}
node goal;
void up(int &x,int k){x=max(x,k);}
node operator+(node v,int t)
{
	if(v==goal)return goal;
	static node ans;ans.clear();
	ans.cnt=v.cnt+(t>=2);
	if(ans.cnt>=7)return goal;
	for(int i=0;i<3;i++)for(int j=0;j<3;j++)for(int k=0;k<2;k++)
	{
		int o=v.f[i][j][k];
		if(o<0)continue;
		for(int a=0;a<=min(i,t);a++)//枚举预留组数 
		for(int b=0;b<=min(j,t);b++)//枚举走几个顺子
		if(a+b<=t)
		{
			for(int c=0;c<=1;c++)//枚举走几个对子
			for(int d=0;d<=1;d++)//枚举走几个面子
			if(a+b+c*2+d*3<=t)up(ans.f[min(2,t-a-b-c*2-d*3)][a][k|c],o+b+d);
		}
	}
	for(int i=0;i<3;i++)for(int j=0;j<3;j++)for(int k=0;k<2;k++)
	{
		ans.f[i][j][k]=min(ans.f[i][j][k],4);
		if(k==1&&ans.f[i][j][k]==4)return goal;
	}
	return ans;
}
node qwq[M];
queue<node>q;
map<node,int>mp;
int size,cnt[N],fac[N],C[N][N],dp[N][M],DP[N][M],nxt[M][10];
void build()
{
	node st;
	st.clear();goal.clear();
	st.f[0][0][0]=0;goal.cnt=-1;
	for(int i=0;i<3;i++)for(int j=0;j<3;j++)for(int k=0;k<2;k++)goal.f[i][j][k]=-mo;
	mp[st]=++size;mp[goal]=++size;q.push(st);
	for(int i=0;i<=4;i++)nxt[2][i]=2;
	while(!q.empty())
	{
		node x=q.front();q.pop();
		qwq[mp[x]]=x;
		for(int i=0;i<=4;i++)
		{
			node to=x+i;
			if(!mp.count(to))mp[to]=++size,q.push(to);
			nxt[mp[x]][i]=mp[to];
		}
	}
}
int ksm(int x,int k)
{
	int ans=1;
	while(k){if(k&1)ans=1ll*ans*x%mo;k>>=1;x=1ll*x*x%mo;}
	return ans;
}
int main()
{
	build();
	int n=read(),ans=0;
	for(int i=1;i<=13;i++){cnt[read()]++;read();}
	fac[0]=C[0][0]=1;
	for(int i=1;i<=4*n;i++)
	{
		C[i][0]=1;fac[i]=1ll*fac[i-1]*i%mo;
		for(int j=1;j<=i;j++)C[i][j]=(C[i-1][j-1]+C[i-1][j])%mo;
	}
	DP[0][1]=1;
	for(int o=1;o<=n;o++)
	{
		for(int i=0;i<=4*o;i++)
		for(int x=1;x<=size;x++)
		dp[i][x]=DP[i][x],DP[i][x]=0;
		for(int i=0;i<=4*o;i++)
		for(int x=1;x<=size;x++)
		{
			if(!dp[i][x])continue;
			for(int k=cnt[o];k<=4;k++)
			{
				int to=nxt[x][k];if(to==2)continue;
				DP[i+k][to]=(DP[i+k][to]+(1ll*C[4-cnt[o]][k-cnt[o]]*dp[i][x]%mo))%mo;
			}
		}
	}
	for(int i=0;i<4*n;i++)
	{
		int tot=0;
		for(int x=1;x<=size;x++)tot=(tot+DP[i][x])%mo;
		tot=1ll*tot*fac[i-13]%mo*fac[4*n-i]%mo;
		ans=(ans+1ll*tot*ksm(fac[4*n-13],mo-2))%mo;
	}
	printf("%d",(ans%mo+mo)%mo);
	return 0;
}
posted @ 2019-04-27 18:30  Creed-qwq  阅读(286)  评论(0编辑  收藏  举报