【[HAOI2011]Problem c】
阿新 • • 發佈:2019-01-02
好題啊
先考慮一些如何判掉無解的情況
我們開一個桶,存一下每個編號有多少個人必須選,之後做一個字尾和,之後我們掃一遍,如果一旦有一個字尾和\(pre[i]\)超過\(n-i+1\)就不合法了,因為我們在\([i,n]\)這裡要安排的人已經超過\(n-i+1\)也就是這個區間的容納範圍了,所以這樣放下去肯定會導致超掉,於是就不合法啦
之後我們搞一個\(dp\),設\(dp[i][j]\)表示從\([i,n]\)這些個編號裡安排了\(j\)個人(不算那些提前預定好的人)的方案數之後帶上一個組合數轉移就好了
程式碼
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define maxn 305 #define re register #define LL long long #define max(a,b) ((a)>(b)?(a):(b)) #define min(a,b) ((a)<(b)?(a):(b)) inline int read() { char c=getchar();re int x=0; while(c<'0'||c>'9') c=getchar(); while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x; } int n,mod,T,m,f; int a[maxn],c[maxn][maxn],dp[maxn][maxn],pre[maxn]; int main() { T=read(); while(T--) { n=read(),m=read(),mod=read(),f=0; int x,y; memset(dp,0,sizeof(dp)),memset(a,0,sizeof(a)),memset(c,0,sizeof(c)),memset(pre,0,sizeof(pre)); for(re int i=1;i<=m;i++) x=read(),y=read(),a[y]++; for(re int i=0;i<=n;i++) c[i][0]=c[i][i]=1; for(re int i=2;i<=n;i++) for(re int j=1;j<i;j++) c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod; for(re int i=n;i;--i) pre[i]=pre[i+1]+a[i]; for(re int i=1;i<=n;i++) if(pre[i]>n-i+1) f=1; if(f) {puts("NO");continue;} putchar('Y'),putchar('E'),putchar('S'); dp[n+1][0]=1; for(re int i=n;i;--i) for(re int j=0;j<=n;j++) { if(!dp[i+1][j]) continue; for(re int k=a[i];k<=n-i+1-pre[i+1]-j;k++) dp[i][j+k-a[i]]=(dp[i][j+k-a[i]]+(LL)dp[i+1][j]*(LL)c[n-m-j][k-a[i]]%mod)%mod; } printf(" %d\n",dp[1][n-m]); } return 0; }