再lrh的帮助下填了好大一个坑…orz lrh
设
pji
表示i步操作将序列染成j色的概率
则:
ans=ΣjΣipji∗i
逆用全期望公式得
ans=ΣjΣxpjxΣi1Σxpjxpji∗i
设 f(S,now) 表示由S起始当前在now状态至终点的期望步数
即 Σi1Σxpjxpji∗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);
}