POJ 3744 Scout YYF I 概率DP+矩阵优化

题意：

一条路上有n个地雷，给出地雷的位置。某人从起点（位置1）出发，走一步的概率是p，走两步的概率是（1-p），问这个人活着。。（嗯活着）走过这条路的概率是多少..

思路：

首先n很小的，<=10。然而路太长了....老长以至于我们O(n)的算法都会超时。。。

首先想到，平安走过，即到达 最后一个地雷 的下一个位置的概率。设为dp[x[n]+1];

然后我们会想到对于任意一个位置来讲，会有dp[i] = p * dp[i-1] + (1-p) * dp[i-2]（这个我都会....）

然后我们再想想会发现，1-dp[x[n]]表示的是不会走到x[n]位置的概率...其实就是走到x[n]+1位置的概率了...丿

剩下就是优化过程啦：对于这样的递推式子：dp[i] = p * dp[i-1] + (1-p) * dp[i-2]

我们发现他是有前两项得出的后一项，没错，就是矩阵优化啦！

我们把整个走路的过程分成n份，每求得是从      上个已经到达的位置（这个位置的概率我们在计算下一个x[i]+1位置的时候，设为概率1，及一定在此位置出发~）           走到    x[i]+1位置     的概率~然后我们根据乘法原理，将所有的结果乘一下，就是最终答案啦~

code：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define eps 1e-9
using namespace std;
//-------------------
struct Matrix
{
    double mat[2][2];
};
Matrix mul(Matrix a,Matrix b)
{
    Matrix ret;
    for(int i=0;i<2;i++)
      for(int j=0;j<2;j++)
      {
          ret.mat[i][j]=0;
          for(int k=0;k<2;k++)
            ret.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
      }
    return ret;
}
Matrix pow_M(Matrix a,int n)
{
    Matrix ret;
    memset(ret.mat,0,sizeof(ret.mat));
    for(int i=0;i<2;i++)ret.mat[i][i]=1;
    Matrix temp=a;
    while(n)
    {
        if(n&1)ret=mul(ret,temp);
        temp=mul(temp,temp);
        n>>=1;
    }
    return ret;
}

const int maxn = 30;
int n, x[maxn];
double p;
int main(){
    while(scanf("%d%lf",&n,&p) != EOF){
        for(int i = 0; i < n; i++){
            scanf("%d",&x[i]);
        }
        sort(x, x+n);
        Matrix tt;
        tt.mat[0][0] = p;
        tt.mat[0][1] = 1-p;
        tt.mat[1][0] = 1;
        tt.mat[1][1] = 0;
        Matrix tmp;
        tmp = pow_M(tt, x[0]-1);
        double ans = (1 - tmp.mat[0][0]);

        for(int i = 1; i < n; i++){
            if(x[i] == x[i-1]) continue;
            tmp = pow_M(tt, x[i]-x[i-1]-1);
            ans *= (1 - tmp.mat[0][0]);
        }
        printf("%.7lf\n",ans);
    }
    return 0;
}

ps：如果和bin神代码一样.....................对我就是抄的bin神代码...因为窝不会啊...