poj3744 (概率DP+矩阵快速幂)

http://poj.org/problem?id=3744

题意：在一条铺满地雷的路上，你现在的起点在1处。在N个点处布有地雷，1<=N<=10。地雷点的坐标范围：[1,100000000].

每次前进p的概率前进一步，1-p的概率前进2步。问顺利通过这条路的概率。就是不要走到有地雷的地方。

 

分析：

安全通过就是走到最右的地雷坐标+1 的位置（安全）

有一道显然的转移方程 dp[i] = dp[i-1]*p + dp[i-2]*(1-p) ;   如果有地雷dp[i] 就为0，这样一直的递推下去可是我们发现地雷的坐标真的太大了，递推超时；然后我们在观察这个式子可以发现这是斐波那契数列，可以想到用矩阵快速幂解决 ，但是如果是出现了地雷，dp[i]就会改变，这样就无法使用快速幂。

N个有地雷的点的坐标为 x[1],x[2],x[3]```````x[N].

我们把道路分成N段：

1~x[1];

x[1]+1~x[2];

x[2]+1~x[3];

`

`

`

x[N-1]+1~x[N].

 

这样每一段只有一个地雷。我们只要求得通过每一段的概率。乘法原理相乘就是答案。

对于每一段，通过该段的概率等于1-踩到该段终点的地雷的概率。

 

就比如第一段 1~x[1].  通过该段其实就相当于是到达x[1]+1点。那么p[x[1]+1]=1-p[x[1]].

但是这个前提是p[1]=1,即起点的概率等于1.对于后面的段我们也是一样的假设，这样就乘起来就是答案了。

 (可以理解为我走到那个位置炸了了，那安全的概率肯定是1-不满足的概率啊)

对于每一段的概率的求法可以通过矩阵乘法快速求出来。

#include<stdio.h>
#include<algorithm>
#include<vector>
using namespace std;

typedef long long ll;

typedef vector<double>vec;
typedef vector<vec >mat;
int n;
double p,T1,T2,pT1,pT2;
mat mul(mat &A , mat &B)
{
    mat C(A.size(),vec(B.size()));

    for(int i= ; i<A.size() ; i++)
    {
        for(int k= ; k<B.size() ; k++)
        {
            if(A[i][k]==) continue;
            for(int j= ; j<B[].size() ; j++)
            {
                if(B[k][j]==) continue;
                C[i][j]=(C[i][j] + A[i][k]*B[k][j]);
            }
        }
    }
    return C;
}
mat qpow(mat A,ll n)
{
    mat B(A.size(),vec(A.size()));
    for(int i= ; i<A.size() ; i++)
    B[i][i]=;
    while(n>)
    {
        if(n&)
        B=mul(B,A);
        A=mul(A,A);
        n>>=;
    }
    return B;
}
double so(int len)
{
    mat A(,vec());
    A[][]=p;A[][]=-p;
    A[][]=;A[][]=;
    A = qpow(A,len-);
    double T=A[][]*p+A[][];
    return T;
}
int x[];
bool vis[];
double dp[];
int main()
{
    while(~scanf("%d%lf",&n,&p))
    {
        int Max=-;
        for(int i= ; i<=n ; i++)
        scanf("%d",&x[i]),vis[x[i]]=,Max=max(Max,x[i]);
        dp[]=;
        dp[]=p;
        if(vis[]) dp[]=;
        if(vis[]) dp[]=;
        for(int i= ; i<=Max+ ; i++)
        {

            dp[i]=dp[i-]*p + dp[i-]*(-p);
            if(vis[i]) dp[i]=;
        }
        printf("%0.7f\n",dp[Max+]);

    }
}