BZOJ 2302: [HAOI2011]Problem c（数学+DP）

题面： bzoj_2302

题解：

令\(dp[i][j]\)表示编号 \(\leq i\)的人有j个的方案数；

\(cnt[i]\)表示编号指定为\(i\)的人数，\(sum[i]\)表示编号可以\(\leq i\)的人数

有转移方程

\[dp[i][j]=\sum_{k=cnt[i]}^{j-i+1}dp[i-1][j-k]\times C_{sum[i]-cnt[i]-(j-k)}^{k-cnt[i]}
\]

P.S. 组合数表示现在有\(sum[i]\)个人，\(cnt[i]\)个人已经确定必须选，\(j-k\)个人已经选完了，在剩下的人中选出\(k-cnt[i]\)个人使其编号为\(i\)

#include<bits/stdc++.h>
using namespace std;
namespace Tzh{
	typedef long long ll;
	const int maxn=310;
	ll dp[maxn][maxn],sum[maxn],cnt[maxn],c[maxn][maxn],n,m,p,T;
	void init(){c[0][0]=1;
		for(int i=1;i<=n;i++){c[i][0]=1;
			for(int j=1;j<=i;j++)
				c[i][j]=(c[i-1][j-1]+c[i-1][j])%p;
		}
		memset(cnt,0,sizeof(cnt));
		memset(sum,0,sizeof(sum));
		memset(dp,0,sizeof(dp));
	}
	void work(){
		scanf("%d",&T);
		while(T--){ int x;
			scanf("%d%d%d",&n,&m,&p);init(); sum[0]=n-m;
			for(int i=1;i<=m;i++) scanf("%*d%d",&x),cnt[x]++;
			for(int i=1;i<=n;i++){
				sum[i]=sum[i-1]+cnt[i];
				if(sum[i]<i){
					puts("NO"); goto end;
				}
			} dp[0][0]=1;
			for(int i=1;i<=n;i++)
				for(int j=sum[i];j>=i;j--)
					for(int k=cnt[i];k<=j-i+1;k++)
						dp[i][j]=(dp[i][j]+dp[i-1][j-k]*c[sum[i]-cnt[i]-j+k][k-cnt[i]]%p)%p;
			printf("YES %lld\n",dp[n][n]); end:;
		}
	}
}
int main(){
	Tzh::work();
	return 0;
}