2554: Color

再lrh的帮助下填了好大一个坑…orz lrh
设 $p_i^j$ 表示i步操作将序列染成j色的概率
则：

$$ans=\Sigma_j\Sigma_ip_i^j*i$$
逆用全期望公式得
$$ans=\Sigma_j\Sigma_xp_x^j\Sigma_i\frac1{\Sigma_xp_x^j}p_i^j*i$$
设 $f(S,now)$ 表示由S起始当前在now状态至终点的期望步数
即 $\Sigma_i\frac{1}{\Sigma_xp_x^j}p_i^j*i$
再令b[i]表示 起始有i个并保证一定能到终点的( 即全集为所有由i个起始至终点的方案)至终点的期望步数
那么 $f(S,now)$ 就可以表示了
再大力手推一波递推式子就好了

#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=n;i++)
using namespace std;
typedef double db;
typedef long long ll;
const int N=1e4+7;
struct f{
    db a,b;f(db a=0.0,db b=0.0):a(a),b(b){}
}d[N];
f operator + (const f& x, const f& y){
    return f(x.a + y.a , x.b + y.b);
}
f operator - (const f& x, const f& y){
    return f(x.a - y.a , x.b - y.b);
}
f operator * (const f& x, db k){
    return f(x.a * k , x.b * k);
}
char s[N];int n,t[26];
db cnt[N];
int main(){
    scanf("%s",s+1);n=strlen(s+1);
    if(n==1){puts("0.0");return 0;}
    d[1]=f(1,0);
    rep(i,1,n-1){
        d[i+1]=f(0,-(db)n*(n-1)/(i*(n-i)));
        d[i+1]=d[i+1]-(d[i-1] * ((db)(i-1)/i));
        d[i+1]=d[i+1]+(d[i]*2);
        d[i+1]=d[i+1]*((db)i/(i+1));
    }
    cnt[1]= - d[n].b / d[n].a;
    rep(i,2,n){
        cnt[i]=cnt[1]*d[i].a + d[i].b;
    }
    rep(i,1,n)t[s[i]-'A']++;
    db ans=0.0;
    rep(i,0,25)if(t[i])ans+=cnt[t[i]]*t[i]/n;
    printf("%.1lf",ans);
}