poj 3744 Scout YYF I 矩阵

#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
const int mod=10000;
const double eps=1e-14;
struct matrix{
    double f[2][2];
};
matrix mul(matrix a,matrix b)
{
    matrix c;
    memset(c.f,0,sizeof(c.f));
    int i,j,k;
    for(k=0;k<2;k++)
    {
        for(i=0;i<2;i++)
        {
            if(a.f[i][k]<eps)continue;
            for(j=0;j<2;j++)
            {
                if(b.f[k][j]<eps)continue;
                c.f[i][j]=(c.f[i][j]+a.f[i][k]*b.f[k][j]);
            }
        }
    }
    return c;
}
matrix pow_mod(matrix a,int b)
{
    matrix s;
    memset(s.f,0,sizeof(s.f));
    for(int i=0;i<2;i++)
        s.f[i][i]=1;
    while(b)
    {
        if(b&1)
            s=mul(s,a);
        a=mul(a,a);
        b=b>>1;
    }
    return s;
}
int main()
{
    int n,a[11];
    double p;
    while(cin>>n>>p)
    {
        int i,j,k;
        for(i=1;i<=n;i++)
            cin>>a[i];
        sort(a+1,a+n+1);
        for(i=2;i<=n;i++)
            if(a[i]-a[i-1]==1)
            break;
        if(i<=n||a[1]==1){printf("%.7lf\n",0.0);continue;}//特殊情况处理
        matrix e,g,h;
        e.f[0][0]=p;e.f[0][1]=1;
        e.f[1][0]=1-p;e.f[1][1]=0;
        g.f[0][0]=1;g.f[1][0]=g.f[0][1]=g.f[1][1]=0;
        a[0]=0;
        for(i=1;i<=n;i++)
        {
            if(a[i]==a[i-1])continue;//雷的位置相同
            h=pow_mod(e,a[i]-a[i-1]-2);
            g=mul(g,h);
            g.f[0][0]=g.f[0][0]*(1-p);
            g.f[0][1]=0;
        }
        printf("%.7lf\n",g.f[0][0]);
    }
    return 0;
}
/*
    方法：
    先将地雷位置排序，如果出现两个连着的雷或者在1位置有雷就输出0
    其余情况用矩阵：
        |f[n] f[n-1]|*|p   1|=|f[n+1] f[n]|
                      |1-p 0|  
    计算从前一个雷+1的位置到后一个雷-1的位置的概率。因为只能跳一步或者两步，所以
    i表示雷的位置，f[i+1]=f[i-1]*(1-p);f[i]=0;，如此连续10次矩阵就出来结果了
*/