bzoj 2302 [HAOI2011]Problem c
阿新 • • 發佈:2019-01-06
題意
給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]個人),
是方案合法的充要條件
然後讓我們來考慮一下怎麼求上面的方案數
首先num[i]為欽定i的數量,sn[i]為num[i]的字首和,sum[i]=sn[i]+n-m,也就是最大能有多少個
在求方案之前,我們先考慮怎麼判無解
那麼就是
時無解
然後呢,定義d[i][j]為前i個位置,安排了j個人的方案數
簡單想想可有轉移式
組合數那一大長串就是當前可動的人和需要的數量
(注意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]);
}
}
}