POJ 3744 Scout YYF I(概率DP+数学)

不考虑地雷，设f[i]为在i位置上的概率。有f[i] = p * f[i-1] + (1 - p) * f[i-2]。

考虑地雷的话分段算就好了，然而这个问题的关键在于怎么求这个f[i]。

开始看见这个用的矩阵快速幂，，，结果T了【话说别人也有这样做没T的，但看了半天不知道怎么优化

所以就打算求f[i]通项【就是靠数学咯

算出来f[n] = f[1] * (1 - (p - 1)^n) / (2 - p)。

然后再用快速幂就过了！！而且是0ms！！！

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;
const int maxn = 15;

int a[maxn],n;
double p;

double mod_pow(double a,int k){
    double res = 1;
    while(k){
        if(k & 1) res = res * a;
        a *= a;
        k >>= 1;
    }
    return res;
}

int main(){
    while(~scanf("%d%lf",&n,&p)){
        for(int i=0;i<n;i++) scanf("%d",&a[i]);
        sort(a,a+n);
        if(a[0] == 1){
            printf("%.7f\n",0);
            continue;
        }
        int k;
        double f = 1;
        for(int i=0;i<n;i++){
            if(i == 0) k = a[i] - 1;
            else k = a[i] - a[i-1] - 1;
            f = f * (1 - mod_pow(p-1,k)) / (2 - p);
            f = f * (1 - p);
        }
        printf("%.7f\n",f);
    }
    return 0;
}