POJ3744 Scout YYF 题解

题目链接

分析：

显然是一道概率DP

令$f_i$表示安全到达第$i$格的概率，在不考虑地雷的情况下，显然有：
$$f_1=1,f_2=p,f_i=p{f}_{i-1}+(1-p)f_{i-2}(i>=3)$$
那加上地雷呢？
注意到 $n<=10$，分开处理不就好了，可以分开计算每两个相邻地雷间的区域，遇到地雷就跳过，即:
$$f_{地雷后}=(1-p)f_{地雷前}$$
但 $a_i<=10^8$，会超时？

---

注意到这是一个二阶线性递推数列，用特征方程，解得：$\sout{f_n=\frac{f_1((p-1{)}^n-1)}{p-2}}$,快速幂求就好了~

---

可以用矩阵快速幂优化，令
$$F_n=\left[\begin{array}{c}{f}_n\\ f_{n-1}\end{array}\right]，G=\left[\begin{array}{cc}p& 1-p\\ 1& 0\end{array}\right]$$
则有
$$F_n=F_{n-1}G$$
即
$$F_n=F_1{G}^{n-1}$$
最后跑快速幂即可
注意特判两个地雷相邻的情况

Code:

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 15;
int n,a[maxn];
double p,f;
struct matrix{
    double num[5][5];
}F;
int read(){
    int x = 0;
    char c = getchar();
    while(c < '0' || c > '9') c = getchar();
    while(c >= '0' && c <= '9') x = x * 10 + (c ^ 48),c = getchar();
    return x;
}
matrix mul(matrix x,matrix y){
    matrix d;
    for(int i = 1; i <= 2; i ++){
        for(int j = 1; j <= 2; j ++){
            d.num[i][j] = 0;
            for(int k = 1; k <= 2; k ++) d.num[i][j] += x.num[i][k] * y.num[k][j];
        }
    }
    return d;
}
matrix qpow(matrix x,int k){
    matrix t = x,d;
    d.num[1][1] = d.num[2][2] = 1,d.num[1][2] = d.num[2][1] = 0;
    while(k){
        if(k & 1) d = mul(d,t);
        k >>= 1,t = mul(t,t);
    }
    return d;
}
int main(){
    while(~scanf("%d%lf",&n,&p)){
        for(int i = 1; i <= n; i ++) a[i] = read();
        sort(a + 1,a + 1 + n);
        f = 1,F.num[1][1] = p,F.num[1][2] = 1 - p,F.num[2][1] = 1,F.num[2][2] = 0;
        for(int i = 1; i <= n; i ++){
            if(a[i] - a[i - 1] == 1){//如果相邻，肯定走不出去
                f = 0;
                break;
            }
            if(i != 1) f *= (1 - p);// 如果不是第一段，当前段第一格的概率为从上一段的最后一格跳两格的概率
            f *= qpow(F,a[i] - a[i - 1] - 2).num[1][1];//f变为当前段最后一格的概率
        }
        printf("%.7f\n",f * (1 - p));//最后一段右边还有一个地雷，做完这题才知道输出double要用%f，这题用%lf会WA！调了好久qwq
    }
    return 0;
}