码迷,mamicode.com
首页 > 其他好文 > 详细

ZOJ--3822(概率dp)

时间:2015-08-13 22:18:19      阅读:195      评论:0      收藏:0      [点我收藏+]

标签:

  题意:一个n行m列的棋盘,每次可以放一个棋子,问要使得棋盘的每行每列都至少有一个棋子 需要的放棋子次数的期望。


思路:

    定义三维的状态,dp[i][j][k]表示用k天占据了i行j列的概率。

    下一天的概率分四种情况,一个是只占据了新的一行,只占据了新的一列,占据了新的一行和一列,并没有占据新的行和列。

    初始化只用初始化dp[1][1][1]=1.0即可,第一个棋子放上肯定占据新的一行和一列,其它赋值为0即可。 

    求期望的时候要每次的概率要减去在k-1天就达到n*m的概率,再乘以天数,相加即可;

    求概率正推,求期望反推,因为定义的状态是概率,所以正推即可。

代码:

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<iostream>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
typedef unsigned long long llu;

const int maxd=50+5;
int dx[]= {0,1,0,-1};
int dy[]= {1,0,-1,0};
///==========================
int n,m;
double dp[maxd][maxd][maxd*maxd];

int main()
{
    int kase;
    scanf("%d",&kase);
    while(kase--)
    {
        mem(dp,0);
        dp[1][1][1]=1;
        scanf("%d%d",&n,&m);
        for(int k=1; k<=n*m; k++)
            for(int i=1; i<=n; i++)
                for(int j=1; j<=m; j++)
                {
                   dp[i+1][j][k+1]+=(dp[i][j][k]*   ((n-i)*j*1.0)/(n*m-k));
                   dp[i][j+1][k+1]+=(dp[i][j][k]*   ((m-j)*i*1.0)/(n*m-k));
                   dp[i+1][j+1][k+1]+=(dp[i][j][k]* (n-i)*(m-j)*1.0/(n*m-k));
                   dp[i][j][k+1]+=(dp[i][j][k]*     (i*j-k)*1.0/(n*m-k));
                }
        double ans=0.0;
        for(int k=1;k<=n*m;k++)
            ans+=(dp[n][m][k]-dp[n][m][k-1])*k;
        printf("%.12lf\n",ans);
    }

    return 0;
}


版权声明:本文为博主原创文章,未经博主允许不得转载。

ZOJ--3822(概率dp)

标签:

原文地址:http://blog.csdn.net/whoisvip/article/details/47251383

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!