【期望】CF850F Rainbow Balls

【题目】
CF
袋子里有$n$种颜色的球，第$i$个颜色有$a_i$个，当袋子里有两种或以上颜色球时，连续取出两个球，并将第二个球涂成第一个球的颜色，再将两个球放回去。求期望时间对$1e9+7$取模。$n\le 2500,a_i\le 10^5$

【解题思路】
设$s=\sum a_i$
首先我们可以枚举最后剩下的颜色，设$f_i$表示当前有$i$个这种颜色球时，所有球都变成这个颜色的期望，那么$f_0$不存在，$f_s=0$。

这个问题实际上可以转化为一个序列上的随机游走问题：一个长度为$s+1$的序列，位置编号$0\sim s$，在$0<i<s$的地方向前后的概率是相等的，在序列的两头会停下来。
这个是一个经典的赌徒破产问题，设状态为当前在某个点到达端点的概率，我们可以将方程表示成常系数线性递推的关系，解特征根方程来做。巧合的是这个问题在考前数学组的同学来问过我然后我随便搞了一下就搞出来了。

由于概率相等比较特殊，我们不妨设从$i$走到$s$的概率为$p_i$，那么我们有$p_0=0,p_s=1,p_i=\frac{p_{i-1}+p_{i+1}}{2}(0<i<s)$，然后$p_{i+1}-p_i=p_i-p_{i-1}$，就是一个等差数列的形式了。

于是不难得到在$i$点走到$s$的概率为$\frac{i}{s}$，那么走到$0$就是$\frac{s-i}{s}$

这样依赖我们就可以列出$f$的方程，设$p$表示选出两个不同色球的概率，那么这里$p=\frac{i(s-i)}{s(s-1)}$：
$$f_i=p(f_{i-1}+f_{i+1})+(1-2p)f_i+\frac{i}{s}(0<i<s)$$

这里显然可以高斯消元做到$O((\sum a_i{)}^3)$

考虑化简这个方程，由于没有$f_0$，我们可以得到：
$$f_1=p{f}_2+(1-2p)f_1+\frac{1}{s}$$
即$f_2=2f_1-1$

又因为（我们将原式移项）：
$$f_i-f_{i+1}=f_{i-1}-f_i+\frac{s-1}{s-i}$$
所以有：
$$f_1=f_1-f_s=\sum _{i=2}^s{f}_{i-1}-f_i=(s-1)(f_1-f_2)+\sum _{i=2}^{s-1}\frac{s-1}{s-i}(s-i)$$
回代得：
$$f_1=\frac{(s-1{)}^2}{s}$$
于是我们可以求出$f_2$，然后就可以递推出所有的$f$了。
最后答案就是${\sum }_{i=1}^n{f}_{a_i}$
复杂度 O ( n + max ⁡ { a i } ) O(n+\max \{a_i\}) O(n+max{ai​})

【参考代码】

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N=1e5+10,mod=1e9+7;
int n,mx,sum,ans;
int a[N],f[N];

int qpow(int x,int y)
{
	int res=1;
	for(;y;y>>=1,x=(ll)x*x%mod)if(y&1)res=(ll)res*x%mod;
	return res;
}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("CF850F.in","r",stdin);
	freopen("CF850F.out","w",stdout);
#endif
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]),mx=max(mx,a[i]),sum+=a[i];
	f[1]=(ll)(sum-1)*(sum-1)%mod*qpow(sum,mod-2)%mod;
	for(int i=1;i<mx;++i) 
		f[i+1]=((ll)f[i]*2-f[i-1]-(ll)(sum-1)*qpow(sum-i,mod-2)%mod+mod*2)%mod;
	for(int i=1;i<=n;++i) ans=(ans+f[a[i]])%mod;
	printf("%d\n",ans);
	return 0;
}