poj 3744 Scout YYF I （概率dp）

题目链接：哆啦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;
}