BZOJ2302: [HAOI2011]Problem c|动态规划|组合数学

可以发现若编号<=i的人数<i则无解

否则就有解

F[i][j]表示编号<=i的人有j个的方案数

s[i]表示编号必须为i的人的个数

sum[i]表示编号可以为<i的人的个数

C[i][j]表示从i个人中选j个人的方案数

突然发现 C[i][j]符合杨辉三角 如此卧槽.....给自己智商跪了

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,m,P;
int s[333],sum[333];
ll C[333][333],f[333][333];
void previous()
{
	int i,j;
	memset(s,0,sizeof(s));
	memset(f,0,sizeof(f));
	memset(C,0,sizeof(C));
	for(i=0; i<=n; i++)
	    for(C[i][0]=1,j=1;j<=i; j++)
	        C[i][j]=(C[i-1][j]+C[i-1][j-1])%P;
}
int main()
{
	int i,T,x;
	for(scanf("%d",&T); T; T--)
	{
		scanf("%d%d%d",&n,&m,&P);
		sum[0]=n-m;
		previous();
		for(i=1; i<=m; i++) scanf("%*d%d",&x),s[x]++;
		for(i=1; i<=n; i++)
		{
			sum[i]=sum[i-1]+s[i];
			if(sum[i]<i)
			{
				puts("NO");
				break;
			}
		}
		if(i!=n+1) continue;
		f[0][0]=1;
		for(int i=1; i<=n; i++)
		    for(int j=sum[i]; j>=i; j--)
		        for(int k=j-i+1; k>=s[i]; k--)
		        f[i][j]+=f[i-1][j-k]*C[sum[i]-j+k-s[i]][k-s[i]],
		        f[i][j]%=P;
		printf("YES %lld\n",f[n][n]);
	}
	return 0;
}