关闭

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

196人阅读 评论(1) 收藏 举报
分类:

不考虑地雷,设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;
}


1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:36696次
    • 积分:2553
    • 等级:
    • 排名:第14819名
    • 原创:228篇
    • 转载:8篇
    • 译文:0篇
    • 评论:1条
    文章分类