爱悠闲 > BZOJ 2302 HAOI2011 Problem c 动态规划

BZOJ 2302 HAOI2011 Problem c 动态规划

分类: BZOJ  |  标签: BZOJ,BZOJ2302,动态规划  |  作者: popoqqq 相关  |  发布日期 : 2015-09-25  |  热度 : 918°

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

一个序列合法等价于编号 i 的人至少有 i
然后就可以DP辣。。。
fi,j 表示编号 i 的人有 j 个的方案数, cnti 表示确定编号为 i 的人的个数, sumi 表示编号可以 i 的人的个数
那么有
fi,j=ji+1k=cntifi1,jkCkcntisumicnti(jk)
那个组合数表示现在有 sumi 个人, cnti 个人已经确定必须选, jk 个人已经选完了,在剩下的人中选出 kcnti 个人使其编号为 i
时间复杂度 O(Tn3)
少打个回车调了一晚上……我真是老了啊……

#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;
}

版权声明:本文为博主原创文章,未经博主允许不得转载。