首页 > 代码库 > poj 3744 Scout YYF I(矩阵优化概率)
poj 3744 Scout YYF I(矩阵优化概率)
http://poj.org/problem?id=3744
有n个雷,某人的起始位置在1,每次走一步的概率为p,走两步的概率是1-p,给出n个雷的位置,问最后成功走出雷区的概率。
放在高中应该是很简单的分步乘法求概率。即把每一个雷都没踩到的概率求出来,最后n个相乘就是顺利通过的概率。对于输入的n个位置进行分段1~num[1],num[1]+1~num[2]......每一段都只有一个雷num[i],每一段内踩不到雷的概率就是1-踩到num[i]雷的概率。
设dp[i]表示踩到第i个雷的概率,那么dp[i] = p*dp[i-1] + (1-p)*dp[i-2],因为i较大,可以用矩阵相乘进行优化。
#include <stdio.h> #include <iostream> #include <map> #include <set> #include <list> #include <stack> #include <vector> #include <math.h> #include <string.h> #include <queue> #include <string> #include <stdlib.h> #include <algorithm> //#define LL __int64 #define LL long long #define eps 1e-12 #define PI acos(-1.0) using namespace std; const int INF = 0x3f3f3f3f; const int maxn = 4010; struct matrix { double mat[2][2]; void init() { mat[0][0] = mat[1][1] = 1; mat[0][1] = mat[1][0] = 0; } }a,b,res; int num[15]; int n; double p; matrix mul(matrix x, matrix y) { matrix t; memset(t.mat,0,sizeof(t.mat)); for(int i = 0; i < 2; i++) { for(int k = 0; k < 2; k++) { if(x.mat[i][k] == 0) continue; for(int j = 0; j < 2; j++) { t.mat[i][j] += x.mat[i][k]*y.mat[k][j]; } } } return t; } matrix pow(matrix a, int n) { matrix t; t.init(); while(n) { if(n&1) t = mul(t,a); n >>= 1; a = mul(a,a); } return t; } int main() { while(~scanf("%d %lf",&n,&p)) { a.mat[0][0] = p; a.mat[0][1] = 1-p; a.mat[1][0] = 1; a.mat[1][1] = 0; memset(b.mat,0,sizeof(b.mat)); b.mat[0][0] = 1; b.mat[1][0] = 0; for(int i = 1; i <= n; i++) scanf("%d",&num[i]); sort(num+1,num+1+n); double ans; res = pow(a,num[1]-1); res = mul(res,b); ans = 1 - res.mat[0][0]; for(int i = 2; i <= n; i++) { res = pow(a,num[i]-num[i-1]-1); res = mul(res,b); ans *= (1-res.mat[0][0]); } printf("%.7f\n",ans); } return 0; }
poj 3744 Scout YYF I(矩阵优化概率)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。