Scout YYF I POJ - 3744（矩阵优化）

题意：一条路上有n个地雷，给出地雷的位置。某人从起点（位置1）出发，走一步的概率是p，走两步的概率是（1-p），然后问有多少概率走过这个雷区。

思路：

只要走过最后一个地雷就代表走过雷区了。

而每到 i 这个地方，无非是前一步和前两步走过来的。那么公式就是dp[ i ]= p*dp[ i-1 ]+dp[ i-2]*(1-p)

这是连续没有地雷的区域的走法。那么有地雷呢？

把有红圈的表示地雷。那么像图上进行分段，将上一段不踩雷的概率就可以当做下一段的开始的概率。

则 dp[ a[i] +1] = 1 - d[ a[i] ]; 那么这样就化成求每一段的概率直接相乘就行了。

优化： 如果直接乘得话，会超时数据量太大。那么，观察dp[ i ]= p*dp[ i-1 ]+dp[ i-2]*(1-p)， 可以很容易得到这是一个类似 a(n) = p*a(n-1)+ q*a(n-1)

的数列， 那么直接矩阵快速幂就行了。

细节：注意，输入的每个地雷的坐标是无序，所以要先对其排序，还有就是，输入了重复的地雷，但是对于幂来说，不需要负数，所以当相同坐标的地雷就不用计算了。

ac代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
struct jz
{
    double num[][];
    jz operator*(const jz &p)const
    {
        jz ans;
        for (int i = ; i < ;++i)
        for (int j = ; j < ; ++j)
        {
            ans.num[i][j] = ;
            for (int k = ; k < ; ++k)
                ans.num[i][j] += num[i][k] * p.num[k][j];
        }
        return ans;
    }
};
jz POW(jz x, int n)
{
    jz ans;
    memset(ans.num, , sizeof(ans.num));
    for (int i = ; i < ; ++i) ans.num[i][i] = ;
    while (n)
    {
        if (n & )ans = ans*x;
        x = x*x;
        n >>= ;
    }
    return ans;
}
int a[];
int main()
{
    int n;
    double p;
    jz dp;
    while (scanf("%d%lf", &n, &p)!=EOF)
    {
        dp.num[][] = p; dp.num[][] = 1.0 - p;
        dp.num[][] = 1.0; dp.num[][] = 0.0;
        for (int i = 1.0; i <= n; ++i)    scanf("%d", a+i);
        sort(a+, a + n+);
        double sum = 1.0;
        for (int i = ; i <= n; ++i)
        {
            if (a[i] == a[i - ])continue;
            jz ans = POW(dp, a[i] - a[i - ] - );
            sum *= (1.0 - ans.num[][]);
        }
        printf("%.7lf\n", sum);
    }
    return ;
}