题目链接:哆啦A梦传送门
题意:一条路上有n个炸弹,每个炸弹的位置为x,一个人从1开始出发, 每次可以走一步或者走两步,走一步的概率为p,走两步的概率为1-p。问:安全通过这条路的概率是多少?
题解:
我们将全场分段走,保证每段的末尾至少有一个炸弹,然后求每一段安全通过的概率。每一段都是独立的,也就是最后通过全程的概率为通过每一段概率的乘积。
我们设dp[i]为安全通过i点的概率,故我们有这样的转移方程:
dp[i]=p*dp[i-1]+(1-p)*dp[i-2]。
但是数据量很大,不能直接一遍扫过去。
我们观察这个式子,发现这个可以用矩阵快速幂求解。
故每一段都可以用矩阵快速幂求解,每一段的初始点dp[1]=1,并且dp[0]=0。
每一段:
1~x[1]
x[1]+1~x[2]
...
我们要求的是通过每一段的概率,也就是dp[x[i]+1],它等价于求1-dp[x[1]]。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
const int N=1010;
struct mat{
double a[2][2];
};
mat mat_mul(mat x,mat y)
{
mat res;
for(int i=0;i<2;i++)
{
for(int j=0;j<2;j++){
res.a[i][j]=0;
for(int k=0;k<2;k++)
res.a[i][j]+=x.a[i][k]*y.a[k][j];
}
}
return res;
}
mat mat_pow(mat item,int n)
{
mat c,res;
c=item;
memset(res.a,0,sizeof(res.a));
for(int i=0;i<2;i++) res.a[i][i]=1;
while(n)
{
if(n&1) res=mat_mul(res,c);
c=mat_mul(c,c);
n>>=1;
}
return res;
}
int x[20];
int main()
{
int n;
double p;
while(~scanf("%d%lf",&n,&p))
// while(cin>>n>>p)
{
for(int i=1;i<=n;i++)
scanf("%d",&x[i]);
sort(x+1,x+1+n);
double sum=1.0;
mat item;
item.a[0][0]=p,item.a[0][1]=1.0-p;
item.a[1][0]=1,item.a[1][1]=0;
///求第一段的概率1-dp[x[1]]
mat tt=mat_pow(item,x[1]-1);
sum=sum*(1-tt.a[0][0]);
for(int i=2;i<=n;i++)
{
if(x[i]==x[i-1]) continue;
tt=mat_pow(item,x[i]-x[i-1]-1);
sum=sum*(1-tt.a[0][0]);
}
printf("%.7f\n",sum); ///poj中G++要%.f,c++%.lf
}
return 0;
}