POJ 3744 Scout YYF I Scout YYF I (递推＋矩阵快速幂)

题意

有n个地方有地雷，初始站在1处，有p的概率往前走1步，1-p的概率往前走2步，求安全通过的概率。

思路

首先我们能想到的是直接递推即dp[i]=p∗dp[i−1]+(1−p)∗dp[i−2]dp[i]=p∗dp[i−1]+(1−p)∗dp[i−2]然后矩阵快速幂加速一下，但是这样做中间的地雷很难处理。 
因为n很小我们就分段求，对每一个地雷求从前一个地雷到通过的概率，比如求通过第x个地雷的概率就是从dp[x−1]+1走到dp[x]+1dp[x−1]+1走到dp[x]+1的概率，因为只有一步和两步，这个概率直接1−dp[x]1−dp[x]就可以了。 
矩阵就是经典的斐波那契递推矩阵。

 

 dp[i] = dp[i-1]*p+dp[i-2]*(1-p);表示抵达第i个格子的概率。转化成矩阵相乘

 

|  p      1-p    |  |    dp[i-1]    |          |   dp[i]     |

|  1      0       |  |    dp[i-2]    |     =   |   dp[i-1]  |

分别求出这个雷区到上一个雷区踩雷的概率，然后求对立事件，就是等于成功越过第一个雷区，成功越过第二个雷区，成功越过第三个雷区。。。。等一系列步骤而完成，根据乘法法则。相乘。

dp[n]              =   ( p   1-p) ^(n-1)    *        dp[1]

dp[n-1]                  1     0                          dp[0]

 

【代码】

#include<bits/stdc++.h>
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;
}

int x[30];
int main()
{
    int n;
    double p;
    while(scanf("%d%lf",&n,&p)!=EOF)//POJ上G++要改为cin输入
    {
        for(int i=0;i<n;i++)
          scanf("%d",&x[i]);
        sort(x,x+n);
        double ans=1;
        Matrix tt;
        tt.mat[0][0]=p;
        tt.mat[0][1]=1-p;
        tt.mat[1][0]=1;
        tt.mat[1][1]=0;
        Matrix temp;
    
        temp=pow_M(tt,x[0]-1);
        ans*=(1-temp.mat[0][0]);

        for(int i=1;i<n;i++)
        {
            if(x[i]==x[i-1])continue;
            temp=pow_M(tt,x[i]-x[i-1]-1);
            ans*=(1-temp.mat[0][0]);
        }
        printf("%.7lf\n",ans);//POJ上G++要改为%.7f
    }
    return 0;
}