当前位置: 代码迷 >> 综合 >> hdu - 4649 - Professor Tian(概率dp)
  详细解决方案

hdu - 4649 - Professor Tian(概率dp)

热度:46   发布时间:2024-01-10 13:27:50.0

题意:一个数学表达式,n+1个操作数,n个操作符(&, |, ^),每个操作符及其后面的那个操作数有一个消失的概率,问表达式的期望(操作数 < 2^20, 0 < n <= 200)。

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4649

——>>操作数的范围 < 2^20,在暗示着此题用位运算——每个操作数可化为最多20位的二进制数,求出n+1个数的运算后每一个二进制位出现1的概率(出现0的概率不用求,若这一位是0,那么总数的大小与这一位无关,只是占位作用)。

设a[i][j]表示第i个数的第j个二进制位;p[i][j]表示对于二进制位的第bit位,前i个数运算后得到j的概率(j为0或1);d[i]表示这n+1个数运算后第i个二进制位出现1的概率。

对于p[i][j],状态转移方程如下:

p[i][1] = p[i-1][1] * P[i] + 各种运算产生的概率。

p[i][0] = 1 - p[i][1]。

#include <cstdio>using namespace std;const int maxn = 200 + 3;
const int maxm = 20 + 3;
int A[maxn], a[maxn][maxm], n;
char O[maxn];
double P[maxn], p[maxn][2], d[maxm];void read(){int i;for(i = 0; i < n+1; i++) scanf("%d", &A[i]);for(i = 1; i <= n; i++){getchar();O[i] = getchar();}for(i = 1; i <= n; i++) scanf("%lf", &P[i]);
}void init(){int i, j;for(i = 0; i < n+1; i++)for(j = 0; j < 20; j++) a[i][j] = (1 << j) & A[i];
}void dp(){int i, bit;for(bit = 0; bit < 20; bit++){if(a[0][bit]){p[0][1] = 1;p[0][0] = 0;}else{p[0][0] = 1;p[0][1] = 0;}for(i = 1; i < n+1; i++){p[i][1] = p[i-1][1] * P[i];switch(O[i]){case '&':{if(a[i][bit]) p[i][1] += p[i-1][1] * (1 - P[i]);break;}case '|':{if(a[i][bit]) p[i][1] += 1 - P[i];else p[i][1] += p[i-1][1] * (1 - P[i]);break;}case '^':{if(a[i][bit]) p[i][1] += p[i-1][0] * (1 - P[i]);else p[i][1] += p[i-1][1] * (1 - P[i]);break;}}p[i][0] = 1 - p[i][1];}d[bit] = p[n][1];}
}double solve(){double ret = 0;for(int i = 0; i < 20; i++) ret += (1 << i) * d[i];return ret;
}int main()
{int kase = 1;while(scanf("%d", &n) == 1){read();init();dp();printf("Case %d:\n%.6lf\n", kase++, solve());}return 0;
}

用个函数调用的写法,装装B~^_^