[ BZOJ 4318 & 3450 / CodeForces 235 B ] OSU!
\(\\\)
\(Description\)
一共进行\(N\)次操作,生成一个长度为\(N\)的\(01\)序列,成功对应\(1\),失败对应\(0\),已知每一次操作的成功率\(p_i\)。
在这个序列中连续且极长的\(X\)个\(1\)可以贡献\(X^2\)的分数,求期望总分。
- \(N\in [1,10^5]\)
\(\\\)
\(Solution\)
-
考虑增量的思路很可以啊。长度平方的期望并不等于期望长度的平方。所以需要直接考虑长度平方的期望变化。
-
当长度从\(X\)增加到\(X+1\),\(\Delta X^2=(X+1)^2-X^2=2X+1\),所以维护答案只需要考虑长度的期望。
-
考虑一次的答案期望\(x_1[i]\)表示到第\(i\)个位置为止的长度期望,有\(x_1[i]=(x_1[i-1]+1)\times p_i\),代表继承上一个为止的期望长度会\(+1\),但保证第\(i\)位合法,需要乘上\(p_i\)。
-
然后到第\(i\)个位置的平方期望就可以转移自第\(i-1\)个位置,注意增量是有概率的。
\[ans[i]=ans[i-1]+(2x_1[i]+1)\times p_i \]
\(\\\)
\(Code\)
#include<cmath>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 300010
#define R register
#define gc getchar
using namespace std;
int n;
double p[N],x1[N],ans[N];
inline double calc(char x){
return x=='x'?0.0:(x=='o'?1.0:0.5);
}
int main(){
scanf("%d",&n);
char c=gc();
while(c!='o'&&c!='x'&&c!='?') c=gc();
p[1]=calc(c);
for(R int i=2;i<=n;++i) p[i]=calc(gc());
for(R int i=1;i<=n;++i){
x1[i]=(x1[i-1]+1)*p[i];
ans[i]=ans[i-1]+(x1[i-1]*2+1)*p[i];
}
printf("%.4lf",ans[n]);
return 0;
}
\(\\\)
\(Extra\)
得分改为\(X^3\),求分数的期望。
\(\\\)
\(Solution\)
-
同样考虑增量,有\(\Delta X^3=(X+1)^3-X^3=X^3+3X^2+3X+1-X^3=3X^2+3X+1\)
-
于是维护\(x_1[i]\)代表长度的期望,\(x_2[i]\)表示长度平方的期望,有:
\[x_1[i]=(x_1[i-1]+1)\times p_i \]\[x_2[i]=x_2[i-1]+2x_1[i]+1\times p_i \] -
更新答案方式相同,有\(ans[i]=ans[i-1]+(3x_2[i]+3x_1[i]+1)\times p_i\)。
-
解释一下\(ans[i-1]\)的部分,它代表的是前一位置的答案,这一位置是否对答案有贡献是有概率的,而增量是后面括起来的部分。
\(\\\)
\(Code\)
#include<cmath>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 100010
#define R register
#define gc getchar
using namespace std;
inline int rd(){
int x=0; bool f=0; char c=gc();
while(!isdigit(c)){if(c=='-')f=1;c=gc();}
while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=gc();}
return f?-x:x;
}
inline double rdd(){
double x=0,base=1; bool f=0; char c=gc();
while(!isdigit(c)){if(c=='-')f=1;c=gc();}
while(isdigit(c)){x=x*10+(c^48);c=gc();}
if(c=='.'){
c=gc();
while(isdigit(c)){x+=(base/=10)*(c^48);c=gc();}
}
return f?-x:x;
}
int n;
double p[N],x1[N],x2[N],ans[N];
int main(){
n=rd();
for(R int i=1;i<=n;++i) p[i]=rdd();
for(R int i=1;i<=n;++i){
x1[i]=(x1[i-1]+1)*p[i];
x2[i]=(x2[i-1]+x1[i-1]*2+1)*p[i];
ans[i]=ans[i-1]+(x2[i-1]*3+x1[i-1]*3+1)*p[i];
}
printf("%.1lf",ans[n]);
return 0;
}