HDU 5781 ATM Mechine (概率DP)

最新推荐文章于 2019-08-21 21:59:25 发布

姜团长

最新推荐文章于 2019-08-21 21:59:25 发布

阅读量389

点赞数

本文链接：https://blog.csdn.net/AngOn823/article/details/52136913

版权

这是一道比较简单的概率dp题目，但难的是特么的这是概率dp，弱没做过啊……

研读完各类题解之后，理解如下：

dp[k][w]表示，上限为k，警告次数还剩w次时的取完所有钱次数的期望

首先，要取完所有钱，总的情况是[0,k]各取一次，所以分母是k+1.假设实际的存款是x

dp[k-y][w]表示，取了y元钱，并且没有被警告，即（x>=y)，在[0,k]里，比y大的数（x）有k+1-y个，所以这种情况的概率是(k+1-y)/(k+1)，设为p1；

dp[y-1][w-1]表示，取了y元钱，但是被警告了，即（x<y），在[0,k]里，比y小的数（x）有y个，所以这种情况的概率是y/(k+1)，设为p2。

所以转移方程就是：dp[k][w]=min(dp[k][w],p1*dp[k-y][w])+p2*dp[y-1][w-1]+1)

【代码】

/* ***********************************************
Author        :angon

************************************************ */
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)
#define REP(i,k,n) for(int i=k;i<n;i++)
#define REPP(i,k,n) for(int i=k;i<=n;i++)
#define scan(d) scanf("%d",&d)
#define scanl(d) scanf("%I64d",&d)
#define scann(n,m) scanf("%d%d",&n,&m)
#define scannl(n,m) scanf("%i64d%I64d",&n,&m)
#define mst(a,k)  memset(a,k,sizeof(a))
#define LL long long
#define N 2005
#define mod 1000000007
inline int read(){int s=0;char ch=getchar();for(; ch<'0'||ch>'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}

double dp[N][12];


int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    REP(i,0,N)  REP(j,0,12) dp[i][j]=1e9;
    for(int i=0;i<12;i++) dp[0][i]=0;  //k=0,时只要0次。
    for(int k=1;k<N;k++)
        for(int w=1;w<12;w++)
            for(int y=1;y<=k;y++)
                dp[k][w] = min(dp[k][w], (dp[k-y][w]*(k-y+1) + dp[y-1][w-1]*y)/(double)(k+1) + 1);
    
    int k,w;
    while(~scann(k,w))
    {
        w = min(11,w);
        printf("%.6lf\n",dp[k][w]);
    }
    return 0;
}