BZOJ 2302 HAOI2011 Problem c 动态规划

题目大意：给定$n$个人和$n$个位置，要求生成一个序列$a_i$，然后第$1...n$个人依次走到第$a_{1...n}$个位置，如果那个位置已经有人了就走到下一个位置，直到找到一个空位，坐下。如果找完第$n$个座位还是没有找到就称这个序列不合法
现在已经确定了一些$a_i$，求合法序列的数量

一个序列合法等价于编号$\leq i$的人至少有$i$个
然后就可以DP辣。。。
令$f_{i,j}$表示编号$\leq i$的人有$j$个的方案数，$cnt_i$表示确定编号为$i$的人的个数，$sum_i$表示编号可以$\leq i$的人的个数
那么有
$f_{i,j}=\sum_{k=cnt_i}^{j-i+1}f_{i-1,j-k}*C_{sum_i-cnt_i-(j-k)}^{k-cnt_i}$
那个组合数表示现在有$sum_i$个人，$cnt_i$个人已经确定必须选，$j-k$个人已经选完了，在剩下的人中选出$k-cnt_i$个人使其编号为$i$
时间复杂度$O(Tn^3)$
少打个回车调了一晚上……我真是老了啊……

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 330
using namespace std;
int n,m,p;
int cnt[M],sum[M];
long long C[M][M],f[M][M];
void Initialize()
{
    int i,j;
    memset(cnt,0,sizeof cnt);
    memset(C,0,sizeof C);
    memset(f,0,sizeof f);
    for(i=0;i<=n;i++)
        for(C[i][0]=1,j=1;j<=i;j++)
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%p;
}
int main()
{
    int T,i,j,k,x;
    for(cin>>T;T;T--)
    {
        cin>>n>>m>>p;
        sum[0]=n-m;
        Initialize();
        for(i=1;i<=m;i++)
            scanf("%*d%d",&x),cnt[x]++;
        for(i=1;i<=n;i++)
        {
            sum[i]=sum[i-1]+cnt[i];
            if(sum[i]<i)
            {
                puts("NO");
                break;
            }
        }
        if(i!=n+1) continue;
        f[0][0]=1;
        for(i=1;i<=n;i++)
            for(j=sum[i];j>=i;j--)
                for(k=j-i+1;k>=cnt[i];k--)
                    (f[i][j]+=f[i-1][j-k]*C[sum[i]-j+k-cnt[i]][k-cnt[i]])%=p;
        printf("YES %d\n",(int)f[n][n]);
    }
    return 0;
}