POJ 3744 Scout YYF I 概率dp 矩阵快速幂

题意：

一个人要通过长度为n的路，路上有一些陷阱。这个人每次有p概率跳1步，有（1-p）概率跳2步，一开始在第1个格子。问你最终能通过这条路的概率是多少。

思路：

参考了kuangbin大神的题解以及http://blog.csdn.net/magicnumber/article/details/6221087

状态定义：

dp[i]：从开始格子到格子i的概率。

转移方程：

dp[i] = dp[i-1]*p+dp[i-2]*(1-p);

————————————————————

因为i已经达到10^8次方，时间空间都超限制。

这里就用到矩阵快速幂。（以后看到比较大的状态数以及比较符合的转移方程，很多都是可以自行YY一个矩阵进行快速幂）

我们把这条路，根据陷阱分为N段。（段段之间互不影响，因为如果要跳过一个陷阱，肯定是从陷阱前一个格子跳两步到陷阱后一个格子，那就说下一段一定是从前一段的陷阱后一格子出发）

进行快速幂，即可得到跳进该段陷阱的概率。

记不能通过第i段的概率为c[i]，则通过了该段的概率就是1-c[i];

最后把通过所有段的概率乘起来(1-c[1])*(1-c[2])*……*(1-c[n])就是题中所要求的。

code：

#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <iomanip>
using namespace std;

typedef long long LL;
const int MAXN = 15;

struct PP
{
    double tp[2][2];
};
int mi[MAXN];
PP mul(PP &t, PP &p)
{
    PP tmp;
    memset(tmp.tp, 0, sizeof(tmp.tp));
    for(int i = 0;i < 2; i++)
        for(int j = 0;j < 2; j++)
            for(int k = 0;k < 2; k++)
                tmp.tp[i][j] += t.tp[i][k] * p.tp[k][j];
    return tmp;
}

double deal(PP &t, int n)
{
    PP tmp = t;
    PP res = (PP){1, 0, 0, 0};
    //PP res = (PP){1, 0, 1, 0};

    while(n)
    {
        if(n & 1)
        {
            res = mul(res, tmp);
        }
        tmp = mul(tmp, tmp);
        n >>= 1;
    }

    return res.tp[0][0];
}

int main()
{
    int n;
    double p;
    ios::sync_with_stdio(false);
    while(cin>>n>>p)
    {
        mi[0] = 0;
        for(int i = 1;i <= n; i++)
        {
            cin>>mi[i];
        }
        sort(mi+1, mi+n+1);

        //
        PP t = (PP){p, 1.0-p, 1, 0};
        double res = 1;
        for(int i = 1; i <= n; i++)
        {
            double tmp = deal(t, mi[i] - mi[i-1] - 1);
            res *= (1-tmp);
        }
        cout<<fixed<<setprecision(7)<<res<<endl;
        //printf("%.7lf\n", res);
    }
    return 0;
}