[联合集训6-11] 期望题

因为$A_1$是必须要减完的，所以我们只要求出对于$A_i(i=2...n)$在$A-1$减完前期望被减了多少次，最后加上$A_1$即可。因为期望的线性性，我们只需要分别考虑每一个$A_i$即可。
不妨以考虑$A_1,A_2$为例。我们可以把其看成二维平面上的一个点$(A_1,A_2)$，每次分别有$\frac{1}{2}$的概率向下/向左移动一个位置，移到坐标轴停止，问在$A_2$方向期望移动了多少步，可以通过枚举终止位置来算。
1. $A_2$取了$i$个时$A_1$已取完，即停在$(0,A_2-i)$的概率$p_i={A_1+i-1\choose i}(\frac{1}{2})^{A_1+i}$，这样的贡献是$i$。
2. $A_2$先取完，这样的概率显然是$1-\sum p_i$ ，贡献是$A_2$。

那么我们把所有的$A_i$扔到一个桶数组里，考虑在$O(\max\{A_i\})$的时间内一并求解所有问题。我们可以从小到大枚举$k$，维护$\sum_{j=0}^k p_j$和$\sum_{j=0}^k j\cdot p_j$，那么对于等于$k+1$的$A_i$的个数$s_{k+1}$，它们的贡献就是$s_{k+1}\cdot ((k+1)(1-\sum p_j)+\sum j\cdot p_j)$。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#define N 100010
#define ll long long
#define up(x,y) (x=(x+(y))%mod)
using namespace std;
const int mod=323232323;
int n,a[N],b[N];
ll fac[N<<1],ifac[N<<1];
ll ksm(ll a,ll b)
{
    ll r=1;
    for(;b;b>>=1,a=a*a%mod)
        if(b&1)r=r*a%mod;
    return r;   
}
ll C(ll a,ll b)
{
    return a<b?0:fac[a]*ifac[b]%mod*ifac[a-b]%mod;
}
int main()
{
    freopen("expectation.in","r",stdin);
    freopen("expectation.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=2;i<=n;i++)
        b[a[i]]++;
    fac[0]=1;
    for(int i=1;i<=200000;i++)
        fac[i]=fac[i-1]*i%mod;
    ifac[200000]=ksm(fac[200000],mod-2);
    for(int i=199999;i>=0;i--)
        ifac[i]=ifac[i+1]*(i+1)%mod;
    ll P=0,Q=0,ans=0,i2=(mod+1)/2;  
    for(int i=0;i<100000;i++)
    {
        ll t=C(a[1]+i-1,i)*ksm(i2,a[1]+i)%mod;
        up(P,t);up(Q,t*i);
        up(ans,((1ll-P+mod)*(i+1)+Q)%mod*b[i+1]);
    }   
    printf("%lld",(ans+a[1])%mod);

    return 0;
}