POJ 2151 Check the difficulty of problems 概率DP

来源:互联网 发布:java时间戳精确到毫秒 编辑:程序博客网 时间:2024/06/05 09:46

题目大意:

就是现在有一场比赛有M道题(<= 30), 一共有T个队伍参赛( <= 1000), 现在知道每只队伍做出每道题的概率,问比赛结束之后所有队伍都有过题且过题最多的队伍过了N题或以上的概率( N <= M)


大致思路:

就是一个典型的dp吧 ( ﹁ ﹁ ), 过程见代码注释

Result  :  Accepted     Memory  :  15996 KB     Time  :  125 ms

/* * Author: Gatevin * Created Time:  2014/12/3 14:44:18 * File Name: Asuna.cpp */#include<iostream>#include<sstream>#include<fstream>#include<vector>#include<list>#include<deque>#include<queue>#include<stack>#include<map>#include<set>#include<bitset>#include<algorithm>#include<cstdio>#include<cstdlib>#include<cstring>#include<cctype>#include<cmath>#include<ctime>#include<iomanip>using namespace std;const double eps(1e-8);typedef long long lint;/* * 用dp[i][j][k]表示第i个队伍在前j道题中做出k道德概率,那么 * dp[i][j][k] = dp[i][j - 1][k - 1]*p[i][j] + dp[i][j - 1][k]*(1 - p[i][j]) 0 < k < j * dp[i][j][k] = dp[i][j - 1][k - 1]*p[i][j] k == j * dp[i][j][k] = dp[i][j - 1][k]*(1 - p[i][j]) k == 0 * 那么这样可以知道每个队伍做出的题目在0~m之间的概率 * 用sp[i]表示第i个队伍做出1 ~ n - 1道题的概率 sp[i] = ∑dp[i][m][1 ~ n - 1] * 用s[i][j]表示前i个队伍中有j个做出了n题或以上的概率 * 则s[i][j] = s[i - 1][j]*sp[i] + s[i - 1][j - 1]*(1 - sp[i] - dp[i][m][0]) 0 < j < i *   s[i][j] = s[i - 1][j]*sp[i] j == 0 *   s[i][j] = s[i - 1][j - 1]*(1 - sp[i] - dp[i][m][0]) j == i * 最后的结果就是∑s[t][1~t] *///这题没有卡精度,看见POJ2151的discuss里面好多人因为.lf 和.f 的问题发了好多帖子,表示想起某只队友...int m, t, n;double dp[1010][31][31];double p[1010][31];double s[1010][1010];double sp[1010];int main(){    while(scanf("%d %d %d", &m, &t, &n), n || t || m)//刚开始n和m打反了交了一发WA...囧    {        for(int i = 1; i <= t; i++)            for(int j = 1; j <= m; j++)                scanf("%lf", &p[i][j]);        memset(dp, 0, sizeof(dp));        for(int i = 1; i <= t; i++)        {            dp[i][0][0] = 1;            for(int j = 1; j <= m; j++)                for(int k = 0; k <= j; k++)                {                    if(k == 0) dp[i][j][k] = dp[i][j - 1][k]*(1 - p[i][j]);                    else if(k == j) dp[i][j][k] = dp[i][j - 1][k - 1]*p[i][j];                    else dp[i][j][k] = dp[i][j - 1][k - 1]*p[i][j] + dp[i][j - 1][k]*(1 - p[i][j]);                    //cout<<dp[i][j][k]<<" dp["<<i<<"]["<<j<<"]["<<k<<"] "<<endl;                }        }        memset(sp, 0, sizeof(sp));        for(int i = 1; i <= t; i++)            for(int k = 1; k < n; k++)                sp[i] += dp[i][m][k];        memset(s, 0, sizeof(s));        s[0][0] = 1;        for(int i = 1; i <= t; i++)            for(int j = 0; j <= i; j++)            {                if(j == 0) s[i][j] = s[i - 1][j]*sp[i];                else if(j == i) s[i][j] = s[i - 1][j - 1]*(1 - sp[i] - dp[i][m][0]);                else s[i][j] = s[i - 1][j]*sp[i] + s[i - 1][j - 1]*(1 - sp[i] - dp[i][m][0]);            }        double ans = 0;        for(int i = 1; i <= t; i++)            ans += s[t][i];        printf("%.3f\n", ans);    }    return 0;}

0 0
原创粉丝点击