码迷,mamicode.com
首页 > 其他好文 > 详细

UNR #3 百鸽笼

时间:2019-07-01 20:06:11      阅读:102      评论:0      收藏:0      [点我收藏+]

标签:def   problem   block   mat   随机   main   cpp   集合   表示   

题目大意:在UOJ管理员群里一共有\(N\)个管理员,为了容纳这些管理员,vfk准备了\(N+1\)个鸽笼。

为了节省空间,vfk把这些鸽笼堆了起来,共有\(n\)列,第i列放了\(a_i\)个鸽笼,满足 \(\sum a_i=N+1\)

每当UR结束,管理员们就会按照编号从小到大的顺序回到鸽笼里,每个管理员回来的时候,会先等概率的在所有还有剩余的鸽笼的列中随机一个列,然后住到这列剩下的鸽笼里编号最小的一个中。

现在\(N\)个管理员都回笼了之后,还有一列会空出一个鸽笼。你能对于每一列,求出这一列有空鸽笼的概率吗?

\(a_i \le 30, N\le 30\)

PKUWC2018猎人杀很像。

不必要的一步,问题可以转化为无限选直到只剩一个\(a_i>0\)

对于每个点,枚举一个集合\(S\),算出这个点在这个集合\(S\)所有点之前被删完的概率。显然集合外的操作我们可以忽略。集合内部的操作我们可以忽略不计。

考虑这个点被选完的时候的操作序列,这个东西可以用一个背包做出来。于是有一个\(2^n * somthing\)的复杂度的东西。我们再加一维表示当前的集合大小,就行了。

瞎写一发,发现复杂度达到了\(O(n^4m^2)\),T了。我们把那个背包做完然后撤销就行了。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=998244353;
inline int add(int a,int b){a+=b;return a>=mod?a-mod:a;}
inline int sub(int a,int b){a-=b;return a<0?a+mod:a;}
inline int mul(int a,int b){return (ll)a*b%mod;}
inline int qpow(int a,int b){int ret=1;for(;b;b>>=1,a=mul(a,a))if(b&1)ret=mul(ret,a);return ret;}
inline int inv(int x){return qpow(x,mod-2);}
/* math */
const int N = 40, SZ = 910;
int n,a[N];
int binom[2010][2010];
inline void preset(int n=2000){
    binom[0][0]=1;
    for(int i=1;i<=n;i++){
        binom[i][0]=binom[i][i]=1;
        for(int j=1;j<i;j++)binom[i][j]=add(binom[i-1][j-1],binom[i-1][j]);
    }
}
int f[N][SZ];
int g[N][SZ];

int sum = 0;
inline void package(int sz){
    for(int i=1;i<=n;i++){
        for(int j=0;j<=sum;++j){
            g[i][j]=f[i][j];
            for(int d=0;d<sz&&j-d>=0;d++){
                g[i][j]=add(g[i][j],mul(f[i-1][j-d],binom[j][d]));
            }
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<=sum;++j){
            f[i][j]=g[i][j];
        }
    }
}

inline void unpackage(int sz){
    for(int i=1;i<=n;i++){
        for(int j=0;j<=sum;++j){
            g[i][j]=f[i][j];
            for(int d=0;d<sz&&j-d>=0;d++){
                g[i][j]=sub(g[i][j],mul(f[i-1][j-d],binom[j][d]));
            }
            f[i][j]=g[i][j];
        }
    }
}


int main()
{
    preset();
    cin >> n;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),sum+=a[i];
    f[0][0]=1;
    for(int i=1;i<=n;i++)package(a[i]);
    for(int i=1;i<=n;i++){
        unpackage(a[i]);
        int ans=0;
        for(int S=1;S<=n;++S){
            for(int len=0;len<=sum;++len){
                int totlen = len+a[i], way = mul(f[S-1][len], binom[len+a[i]-1][a[i]-1]);
                int p=qpow(inv(S),totlen);
                if(S&1) ans=add(ans, mul(p,way));
                else ans=sub(ans, mul(p,way));
            }
        }
        package(a[i]);
        printf("%d ",ans);
    }
    puts("");
}

UNR #3 百鸽笼

标签:def   problem   block   mat   随机   main   cpp   集合   表示   

原文地址:https://www.cnblogs.com/weiyanpeng/p/11116186.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!