1. 程式人生 > >bzoj 2302 [HAOI2011]Problem c

bzoj 2302 [HAOI2011]Problem c

題意

給n個人安排座位,先給每個人一個1~n的編號,設第i個人的編號為ai(不同人的編號可以相同),接著從第一個人開始,大家依次入座,第i個人來了以後嘗試坐到ai,如果ai被佔據了,就嘗試ai+1,ai+1也被佔據了的話就嘗試ai+2,……,如果一直嘗試到第n個都不行,該安排方案就不合法。然而有m個人的編號已經欽定了,你只能安排剩下的人的編號,求有多少種合法的安排方案。對M取模
100%的資料滿足:1≤T≤10,1≤n≤300,0≤m≤n,2≤M≤109,1≤pi、qi≤n 且保證pi互不相同。

題解

讓我們對這個安排座位進行更深的分析
可以發現,(假如選擇i的有z[i]個人),

i , j = 1 i i
\forall i,\sum_{j=1}^i \geq i 是方案合法的充要條件
然後讓我們來考慮一下怎麼求上面的方案數
首先num[i]為欽定i的數量,sn[i]為num[i]的字首和,sum[i]=sn[i]+n-m,也就是最大能有多少個
在求方案之前,我們先考慮怎麼判無解
那麼就是 i ,
s u m [ i ] < i \exist i,sum[i]<i
時無解
然後呢,定義d[i][j]為前i個位置,安排了j個人的方案數
簡單想想可有轉移式
d [ i ] [ j ] = n u m [ i ] j d [ i 1 ] [ j k ] C ( n m ( j k ) + s n [ i 1 ] , [ k n u m [ i ] ] ) d[i][j]=\sum_{num[i]}^jd[i-1][j-k]C(n-m-(j-k)+sn[i-1],[k-num[i]])
組合數那一大長串就是當前可動的人和需要的數量
(注意C(0,0)=1)
至於組合數怎麼求,儘管M是任意的,但是我們並不需要用上lucas,直接楊輝三角即可
注意不能不預先判無解,而是根據d[n][n]是否為0判斷哦
以上。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=305;
int n,m;
int mod;
int d[N][N];
int num[N],sn[N];
int sum[N];
int C[N][N];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        memset(num,0,sizeof num);
        scanf("%d%d%d",&n,&m,&mod);
        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]+C[i-1][j-1])%mod;
        }
        for(int i=1;i<=m;i++){
            int x;
            scanf("%*d%d",&x);
            num[x]++;
        }
        sum[0]=n-m;
        for(int i=1;i<=n;i++)
            sum[i]=sum[i-1]+num[i],
            sn[i]=sn[i-1]+num[i];
        bool nol=false;
        for(int i=1;i<=n;i++)
            if(sum[i]<i){
                nol=true;
                puts("NO");
                break;
            }
        if(!nol){
            memset(d,0,sizeof d);
            d[0][0]=1;
            for(int i=1;i<=n;i++)
                for(int j=max(num[i],i);j<=sum[i];j++){
                    for(int k=num[i];k<=j;k++){
                        d[i][j]=(d[i][j]+1ll*d[i-1][j-k]*C[n-m-(j-k)+sn[i-1]][k-num[i]]%mod)%mod;
                    }
                }
            printf("YES %d\n",d[n][n]);
        }
    }
}