轮流开枪打一个环上的人 , 每次\(p\)的概率打死 , \(p\)始终相同 , 从第\(1\)个人开始 , 求第\(k\)个人成为唯一幸存者的概率
\(19.3.30\)
官方题解先递推出\(f[n]\) , \(f[1]\)用到\(f[n]\) , 套上一个\(n\)的循环 , 总共是\(O(n^2)\)的
设\(f1[i]\)表示\([1,k-1]\)在\(i\)轮以内全死的概率 , \(f2[i]\)表示\([k+1,n]\)在\(i\)轮以内全死的概率 ,
\(s[i]\)表示某一个人在\(i\)轮以内死掉的概率 , 易知\(s[i]=1-(1-p)^i\)
\(f1[i]=s[i]^{(k-1)}\)
\(f2[i]=s[i]^{(n-k)}\)
枚举第\(k\)个人第\(i\)轮死 , \(ans=\sum{f1[i]*f2[i-1]*(s[i]-s[i-1])}\)
\(10^4\)个人枚举\(10^6\)轮就差不多了 , \(f1\)和\(f2\)的预处理是\(O(nl_{og}n)\)的
\(19.4.4\)
首先可以容易地得出某一个人在\(i\)轮以内死掉的概率\(s[i]=1-(1-p)^i\) , 以及在第\(i\)轮死掉的概率\(s[i]-s[i-1]\)
也许就能推出连续一段人在\(i\)轮死掉的概率\(s[i]^{r-l+1}\)
考虑怎么枚举比较方便 : 枚举自己在第\(i\)轮死 ,
那么在此之前其他的都要死 , 即前面的要在\(i\)轮内死 , 后面的要在\(i-1\)轮内死
枚举自己的状态 , 考虑前面和后面要满足的条件 , 参考[JSOI2018]机器人
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+7;
int n,k;
double p,sp,ans,s[N],pw1[N],pw2[N];
double qpow(double a,int b)
{
double ret=1;
while(b)
{
if(b&1)ret*=a;
a*=a,b>>=1;
}
return ret;
}
int main()
{
cin>>p>>n>>k;
if(n==1){cout<<1;return 0;}
for(int i=1;i<=1e5;i++)s[i]=1-qpow(1-p,i);
for(int i=1;i<=1e5;i++)pw1[i]=qpow(s[i],k-1),pw2[i]=qpow(s[i],n-k);
for(int i=1;i<=1e5;i++)
if(k==n)ans+=pw1[i]*(s[i]-s[i-1]);
else if(k==1)ans+=pw2[i-1]*(s[i]-s[i-1]);
else ans+=pw1[i]*pw2[i-1]*(s[i]-s[i-1]);
printf("%0.9f",ans);
}