解析

二倍相鄰項(xiàng)，就要想裂項(xiàng)

純數(shù)學(xué)題
一道黑色的小凱的疑惑

設(shè)總球數(shù)為 $s$
我們先欽定一種顏色留到最后，那么其他顏色可以等價(jià)考慮
設(shè) $f_i$ 為欽定的顏色的球有 $i$ 個(gè)，涂完需要的期望次數(shù)
則有：
$$f_i=(f_{i?1}+f_{i+1})\times p+f_i\times (1?2p)+w_i$$
其中 $p$ 表示有序取出兩個(gè)不同色球的概率，$w_i$ 表示有 $i$ 個(gè)球的顏色留到最后的概率。
不難發(fā)現(xiàn) $p=\frac{i(s?i)}{s(s?1)}$，考慮如何求 $w_i$。
首先有： $$w_0=0,w_s=1$$。
和 $f$ 的轉(zhuǎn)移類似的，有：$$w_i=(w_{i?1}+w_{i+1})\times p+w_i\times (1?2p)$$
化簡移項(xiàng)，得：
$$w_{i+1}?w_i=w_i?w_{i?1}$$
也就是說 $w$ 是一個(gè)等差數(shù)列，那么顯然就有：
$$w_i=\frac{i}{s}$$
把 $p,w$ 帶回原來的轉(zhuǎn)移式：
$$f_i=(f_{i?1}+f_{i+1})\times \frac{i(s?i)}{s(s?1)}+f_i\times (1?\frac{2i(s?i)}{s(s?1)})+\frac{i}{s}$$
化簡，得：
$$2f_i=f_{i+1}+f_{i?1}+\frac{s?1}{s?i}$$
也就是：
$$f_i?f_{i+1}=f_{i?1}+f_i+\frac{s?1}{s?i}$$
把 $i=1$ 帶入，有：
$$f_2=2f_1+1$$
又有：
$$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}\times (s?i)$$
后面那個(gè)和式是由于對(duì)于一個(gè) $\frac{s?1}{s?i}$，它會(huì)產(chǎn)生 $s?i$ 次貢獻(xiàn)。
把 $f_2=2f_1+1$ 帶入，得到：
$$f_1=\frac{(s?1{)}^2}{s}$$
得到 $f_1$ 之后，后面順推即可。
過程中快速冪求逆元，時(shí)間復(fù)雜度 $O(n\log \mathrm{m}\mathrm{o}\mathrm{d})$。

代碼

（式子推完代碼幾乎沒有任何難度了…）

#include<bits/stdc++.h> using namespace std; #define ll long long #define ull unsigned long long #define debug(...) fprintf(stderr,__VA_ARGS__) const int N=2e5+100; const int mod=1e9+7; inline ll read(){ll x(0),f(1);char c=getchar();while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}while(isdigit(c)){x=(x<<1)+(x<<3)+c-'0';c=getchar();}return x*f; } int n,m;inline ll ksm(ll x,ll k){ll res(1);while(k){if(k&1) res=res*x%mod;x=x*x%mod;k>>=1;}return res; } ll a[N]; ll s,f[N],mx;signed main() { #ifndef ONLINE_JUDGE//freopen("a.in","r",stdin);//freopen("a.out","w",stdout); #endifn=read();for(int i=1;i<=n;i++) a[i]=read(),s+=a[i],mx=max(mx,a[i]);f[1]=(s-1)*(s-1)%mod*ksm(s,mod-2)%mod;f[2]=(2*f[1]-1+mod)%mod;for(int i=2;i<mx;i++) f[i+1]=(2*f[i]-f[i-1]+mod-(s-1)*ksm(s-i,mod-2)%mod+mod)%mod;ll ans(0);for(int i=1;i<=n;i++) (ans+=f[a[i]])%=mod;printf("%lld\n",ans);return 0; } /* */

總結(jié)

以上是生活随笔為你收集整理的CF850F Rainbow Balls（数学、期望）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。