1. 程式人生 > >【[HAOI2011]Problem c】

【[HAOI2011]Problem c】

好題啊

先考慮一些如何判掉無解的情況

我們開一個桶,存一下每個編號有多少個人必須選,之後做一個字尾和,之後我們掃一遍,如果一旦有一個字尾和\(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;
}