分组游戏，2019NOI金牌营1第一题，困难的Dp

最新推荐文章于 2019-07-16 11:13:11 发布

Deep_Kevin

最新推荐文章于 2019-07-16 11:13:11 发布

阅读量339

点赞数

分类专栏：思维好题

本文链接：https://blog.csdn.net/Deep_Kevin/article/details/94408509

版权

思维好题专栏收录该内容

12 篇文章 0 订阅

订阅专栏

正题

题目传送门

对我来说太困难了，居然是签到题。

很容易我们可以先排个序，但是很容易想偏成从前往后做Dp。

不妨从后往前，我们考虑 f[i][j] 表示后面i种ai分组后剩下j个人的方案数。

那么答案就是 f[1][0] 。

转移考虑新加入的 i-1 有多少个人，加上前面剩余的j个人，总共就有 t[i-1]+j 个人。

接着我们枚举一下有k个人现在开始组成一队，就一共有 k*(i-1) 个人分成k组。

当然这个 k*(i-1)<=j+t[i-1] 。

首先方案数然后把 k*(i-1) 个人分成k组这东西可以预处理出来。

在乘上 $C_{j+t[i-1]}^{k(i-1)}$ 就可以了。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;

const int maxn=2010;
long long f[maxn][maxn],g[maxn][maxn],c[maxn][maxn],t[maxn];
const long long mod=998244353;
long long n;
int x;

int main(){
    scanf("%lld",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&x),t[x]++;
    c[0][0]=1;
    for(int i=1;i<=n;i++){
        c[i][0]=1;
        for(int j=0;j<=i;j++) c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
    }
    for(int j=1;j<=n;j++){
        g[0][j]=1;
        for(int i=j;i<=n;i+=j) g[i][j]=g[i-j][j]*c[i-1][j-1]%mod;
    }
    f[n+1][0]=1;
    for(int i=n+1;i>1;i--)
        for(int j=0;j<=n;j++)
            for(int k=0;k*(i-1)<=j+t[i-1];k++)
                (f[i-1][j+t[i-1]-k*(i-1)]+=f[i][j]*c[j+t[i-1]][k*(i-1)]%mod*g[k*(i-1)][i-1]%mod)%=mod;
    printf("%lld\n",f[1][0]);
    return 0;
}