[CF850F]Rainbow Balls / [DarkBZOJ2554]Color

题目

传送门 to luogu

传送门 to DarkBZOJ

思路

第一步转化：分类讨论。每个颜色提供的贡献是独立可加的。

枚举某种颜色最后胜利，则其余颜色可视为相同。于是只需要求出该颜色的数量为 $k$ 时的信息。

首先令 $f(x)$ 表示结束的概率。注意到该颜色减少和增加的概率是相同的，都是 $\frac{x(n-x)}{n(n-1)}$，因此必定有方程式

显然，这就是数量增加 $1$ （与减少 $1$ ）的概率，写出递推方程式
$$f(x)=\frac{f(x-1)+f(x+1)}{2}$$

这个式子是等差数列的另一种表示方法，结合边界条件 $f(0)=0,f(n)=1$ 可知
$$f(x)=\frac{x}{n}$$

然后令 $g(x)$ 表示期望。令 $p_x=\frac{x(n-x)}{n(n-1)}$，则
$$g(x)=(1-2p_x)g(x)+p_x[g(x-1)+g(x+1)]+f(x)$$

这与上面的式子是等价的。注意是 $+f(x)$，尽管这是常识，但容易忘。

方程可以变形为
$$2g(x)-g(x-1)-g(x+1)=\frac{f(x)}{p_x}\text{\hspace{1em}(1)}$$

边界是 $g(0)=g(n)=0$ 。这玩意儿转移有环；怎么解方程？

生成函数

记 $H(x)={\sum }_{k⩾1}\frac{f(k)}{p_k}{x}^{k+1}$，显然答案的生成函数为
$$G(x)=\frac{-H(x)+xg(1)}{(1-x{)}^2}$$

我们只需解 $g(1)$ 使得 $[x^n]G(x)=0$ 。

$H(x)$ 没有闭形式，我们就不要再考虑 $\mathrm{GF}$ 了，回到普通计算数学上面来。
$$[x^n]G(x)=ng(1)-\sum _{i=k}^{n-1}\frac{f(k)}{p_k}(n-k)=ng(1)-(n-1{)}^2$$

于是解出了 $g(1)=\frac{(n-1{)}^2}{n}$，就知足吧。

花絮：我非要拼凑 $H(x)$ 闭形式，因此将 $G(x)$ 翻转并递推，就会解出 $g(n-1)=n(H_n-1)$，其中 $H_n={\sum }_{j=1}^n\frac{1}{j}$ 是调和级数，没有快速求解的方法。

初等方法

称 $(1)$ 式为关于 $x$ 的方程。

设 $1⩽l⩽r⩽n$ 。对所有 $k\in [l,r]$，将 $k$ 的方程求和（等式左右分别相加），则有
$$g(l)+g(r)-g(l-1)-g(r+1)=\sum _{k=l}^r\frac{f(k)}{p_k}$$

令 $l=1$ 就得到
$$g(1)+g(r)=g(0)+g(r+1)+\sum _{k=1}^r\frac{f(k)}{p_k}$$

联立 $g(0)=0$ 就有

$$g(x)=g(x+1)-g(1)+\sum _{i=1}^k\frac{f(k)}{p_k}$$

注意到它描述的其实是 $\Delta g$ 。那么求和容易得到
$$g(x)=g(n)-(n-x)g(1)+\sum _{i=x}^{n-1}\sum _{j=1}^i\frac{f(j)}{p_j}$$

取 $x=1$ 后化简式子可知 $g(1)=\frac{(n-1{)}^2}{n}$ 。

代码

这里放出 CF850F Rainbow Balls 的代码。复杂度可以做到线性，但没必要。

#include <cstdio>
#include <algorithm> // Almighty XJX yyds!!
#include <cstring> // oracle: ZXY yydBUS!!!
#include <cctype> // rainybunny root of the evil.
using llong = long long;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
# define rep0(i,a,b) for(int i=(a); i!=(b); ++i)
inline int readint(){
    int a = 0, c = getchar(), f = 1;
    for(; !isdigit(c); c=getchar()) if(c == '-') f = -f;
    for(; isdigit(c); c=getchar()) a = a*10+(c^48);
    return a*f;
}

const int MAXA = 100000, MOD = 1e9+7;
inline llong qkpow(llong b, int q){
    llong a = 1;
    for(; q; q>>=1,b=b*b%MOD) if(q&1) a = a*b%MOD;
    return a;
}

const int MAXN = 2500;
int a[MAXN+1], inv[MAXA+1];

int g[MAXA+1];
int main(){
    int n = readint(), s = 0;
    rep(i,1,n) s += (a[i] = readint());
    rep(i,0,MAXA) inv[i] = int(qkpow(s-i,MOD-2));
    g[0] = 0, g[1] = int(llong(s-1)*(s-1)%MOD*inv[0]%MOD);
    rep0(i,1,MAXA){
        g[i+1] = int((llong(g[i]<<1)-g[i-1]
            -llong(s-1)*inv[i])%MOD);
        if(g[i+1] < 0) g[i+1] += MOD;
    }
    int ans = 0;
    rep(i,1,n) ans = (ans+g[a[i]])%MOD;
    printf("%d\n",ans);
    return 0;
}