Neutralzz的博客

我有自己的梦想和追求!

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;
}


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_26572969/article/details/46825971
个人分类: DP
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭