[cf1479E]School Clubs

对于当前班级状态$S$，定义一个函数$\varphi(S)$，要求其满足：

令结束状态为$S_{end}$，对于任意$S\ne S_{end}$，若其下一个状态为$S'$，则$E(\varphi(S)-\varphi(S'))=1$

由此，归纳即可得到$S$的期望结束步数为为$\varphi(S)-\varphi(S')$

对于一个状态$S$，注意到其与班级编号无关，因此不妨假设其第$i$个班级有$a_{i}$个人（共$m$个班级），再定义一个函数$g(x)$，令$\varphi(S)=\sum_{i=1}^{m}g(a_{i})$

由于为空的班级不影响状态，因此$g(0)=0$

更进一步的，根据$E(\varphi(S)-\varphi(S'))=1$，可得
$$
\frac{1}{2}\sum_{i=1}^{m}\frac{a_{i}}{n}(\Delta_{g}(a_{i}-1)-g(1))+\frac{1}{2}\sum_{i=1}^{m}\sum_{j=1}^{m}\frac{a_{i}a_{j}}{n^{2}}(\Delta_{g}(a_{i}-1)-\Delta_{g}(a_{j}-[i=j]))=1
$$
（其中$\Delta_{g}(x)=g(x+1)-g(x)$）

后者在$i=j$时为0，在$i\ne j$时将两个枚举拆开，即
$$
\frac{1}{2}\sum_{i=1}^{m}\frac{a_{i}}{n}(\Delta_{g}(a_{i}-1)-g(1))+\frac{1}{2}\sum_{i=1}^{n}\frac{a_{i}(n-a_{i})}{n^{2}}(\Delta_{g}(a_{i}-1)-\Delta_{g}(a_{i}))=1
$$
将其整理后，即
$$
\sum_{i=1}^{m}\frac{a_{i}}{2n}[(2-\frac{a_{i}}{n})\Delta_{g}(a_{i}-1)-(1-\frac{a_{i}}{n})\Delta_{g}(a_{i})-g(1)]=1
$$
当对于每一个$a_{i}$，中括号内的值恰为$2$时，根据$\sum_{i=1}^{m}a_{i}=n$，即成立

令$g(1)=-2$，再化简后即可得到$\Delta_{g}(x)$
$$
\Delta_{g}(x)=\frac{2n-x}{n-x}\Delta_{g}(x-1)=\prod_{i=1}^{x}\frac{2n-i}{n-i}\Delta_{g}(0)
$$
根据$g(0)=0$以及$g(1)=-2$，可得初始状态$\Delta_{g}(0)=-2$

由此，可得$g(x)=g(0)+\sum_{i=0}^{x-1}\Delta_{g}(i)=-2\sum_{i=0}^{x-1}\prod_{j=1}^{i}\frac{2n-j}{n-j}$，问题即求$\sum_{i=1}^{m}g(a_{i})-g(n)$

之后暴力$o(n)$计算即可，为了避免每一次求逆，需要用分数的形式计算，以及使用”GNU G++17 9.2.0(64 bit,msys 2)​“的编译器即可通过

标算做到了$o(\sqrt{n}\log n+m\sqrt{n})$的复杂度，可以看cf上的题解

#include<bits/stdc++.h>
using namespace std;
#define M 1005
#define mod 998244353
#define ll long long
int n,m,x,ans,a[M];
ll qpow(int n,int m){
    ll s=n,ans=1;
    while (m){
        if (m&1)ans=ans*s%mod;
        s=s*s%mod;
        m>>=1;
    }
    return ans;
}
ll g(int x){
    ll s1=1,s2=1,ans1=0,ans2=1;
    for(int i=1;i<=x;i++){
        ans1=(s1*ans2+s2*ans1)%mod;
        ans2=s2*ans2%mod;
        s1=s1*(2*n-i)%mod;
        s2=s2*(n-i)%mod;
    }
    return (mod-2)*ans1%mod*qpow(ans2,mod-2)%mod;
}
int main(){
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d",&a[i]);
        n+=a[i];
    }
    for(int i=1;i<=m;i++)ans=(ans+g(a[i]))%mod;
    printf("%d",(ans+mod-g(n))%mod);
}