bzoj2302 [HAOI2011]Problem c

题目链接：bzoj2302
题目大意：
给n个人安排座位，先给每个人一个1~n的编号，设第i个人的编号为ai（不同人的编号可以相同），接着从第一个人开始，大家依次入座，第i个人来了以后尝试坐到ai，如果ai被占据了，就尝试ai+1，ai+1也被占据了的话就尝试ai+2，……，如果一直尝试到第n个都不行，该安排方案就不合法。然而有m个人的编号已经确定(他们或许贿赂了你的上司…)，你只能安排剩下的人的编号，求有多少种合法的安排方案。由于答案可能很大，只需输出其除以M后的余数即可。

题解：
dp+组合数
一开始自己YY了一个状态$f[i][j]$表示编号大于等于i的有j人。但是我想不到转移方程啊QwQ
膜了膜题解之后，发现其实状态表达类似，$f[i][j]$表示编号小于等于i的有j个人。但是还有个数组，$sum[i]$，表示编号可以小于i的有多少个人。啊挺关键的。
由题意易知，若第i个人之前有超过n-ai+1个人的编号大于ai那么方案不合法。换句话说，就是编号小于等于ai的人的个数小于ai，那么方案不合法。
于是不合法的条件判断就是$sum[i]<i$。
然后我们来转移，枚举编号恰好为i的有k个。设$s[i]$表示编号一定为i的人的个数。那么

$$f[i][j]=\sum_{k=s[i]}^{j-i+1}f[i-1][j-k]*C_{sum[i]-s[i]-(j-k)}^{k-s[i]}$$
要恰好有k个i，那么就从能小于等于i的sum[i]个人中选，又有固定了的和j-k个是要小于等于i-1的，所以实际上是从sum[i]-s[i]-(j-k)个人中选k-s[i]个使它们的编号为i的方案数。

记得mod。还有long long

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
#define N 310

LL f[N][N],C[N][N],mod;
int sum[N],s[N];
//sum[i] a可以小于等于i的个数  s[i] 固定要选i的个数
void init()
{
    memset(C,0,sizeof(C));
    for (int i=0;i<=300;i++) C[i][0]=1;
    for (int i=1;i<=300;i++)
     for (int j=1;j<=i;j++)
      C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
}
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int T,n,m,p,q,i,j,k;
    scanf("%d",&T);

    while (T--)
    {
        scanf("%d%d%lld",&n,&m,&mod);
        sum[0]=n-m;init();
        memset(f,0,sizeof(f));
        memset(s,0,sizeof(s));
        for (i=1;i<=m;i++)
        {
            scanf("%d%d",&p,&q);
            s[q]++;
        }
        for (i=1;i<=n;i++)  
        {
            sum[i]=sum[i-1]+s[i];
            if (sum[i]<i) break;
        }
        if (i!=n+1) 
        {
            printf("NO\n");
            continue;
        }
        f[0][0]=1;
        for (i=1;i<=n;i++)
         for (j=sum[i];j>=0;j--)
          for (k=s[i];k<=j-i+1;k++)
          {
              f[i][j]+=f[i-1][j-k]*C[sum[i]-s[i]-(j-k)][k-s[i]]%mod;
              f[i][j]%=mod;
          }
        printf("YES %lld\n",f[n][n]);
    }
    return 0;
}