【XSY1599】dices 期望DP

题目大意

　　Yahtzee是一个非常有名的骰子游戏。有$5$个骰子，$14$个盒子。游戏共有$14$轮，每一轮玩家要投这$5$个骰子，并将这$5$个骰子放到一个没有被用过的盒子内计分。这五个骰子是一样的，每个面朝上的概率是不同的。问你在最优策略下，你能得到的期望得分。

　　计分方式如下：
　　

题解

　　这题题意没有明显注明是先投完骰子再选盒子，导致一些人看错题了。

　　我们设$f_i$为盒子的状态为$i$时最优策略下的期望得分。我们枚举投出来的结果，选择放进哪个盒子中。

　　有时候两种结果本质上是相同的，那么就可以把这两种结果合并。

　　时间复杂度：$O(???)$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#include<utility>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
int a[10];
int b[10];
int c[10];
double p[10];
double q[10000];
double g[10000][20];
double f[1<<15];
int d[20];
double h[20];
int sum;
int ck[10000];
int code()
{
    int s=0;
    int i;
    for(i=1;i<=5;i++)
        s=s*6+b[i]-1;
    return s;
}
int calc(int x)
{
    double s=0;
    int i;
    if(x>=1&&x<=6)
    {
        for(i=1;i<=5;i++)
            if(b[i]==x)
                s+=x;
    }
    else if(x==7)
    {
        if(b[1]==b[2])
        {
            if(b[3]==b[4]&&b[2]!=b[3])
                s=sum;
            else if(b[4]==b[5]&&b[2]!=b[4])
                s=sum;
        }
        else if(b[2]==b[3]&&b[4]==b[5]&&b[3]!=b[4])
            s=sum;
    }
    else if(x==8)
    {
        if(b[1]==b[2]&&b[2]==b[3])
            s=sum;
        else if(b[2]==b[3]&&b[3]==b[4])
            s=sum;
        else if(b[3]==b[4]&&b[4]==b[5])
            s=sum;
    }
    else if(x==9)
    {
        if(b[1]==b[2]&&b[2]==b[3]&&b[3]==b[4])
            s=sum;
        else if(b[2]==b[3]&&b[3]==b[4]&&b[4]==b[5])
            s=sum;
    }
    else if(x==10)
    {
        if(b[1]==b[2]&&b[2]==b[3]&&b[3]!=b[4]&&b[4]==b[5])
            s=25;
        else if(b[1]==b[2]&&b[2]!=b[3]&&b[3]==b[4]&&b[4]==b[5])
            s=25;
    }
    else if(x==11)
    {
        if(c[1]&&c[2]&&c[3]&&c[4])
            s=30;
        else if(c[2]&&c[3]&&c[4]&&c[5])
            s=30;
        else if(c[3]&&c[4]&&c[5]&&c[6])
            s=30;
    }
    else if(x==12)
    {
        if(c[1]&&c[2]&&c[3]&&c[4]&&c[5])
            s=40;
        else if(c[2]&&c[3]&&c[4]&&c[5]&&c[6])
            s=40;
    }
    else if(x==13)
    {
        for(i=1;i<=6;i++)
            if(c[i]==5)
            {
                s=50;
                break;
            }
    }
    else if(x==14)
        s=sum;
    return s;
}
void dfs(int x,double now)
{
    if(x>5)
    {
        int i;
        memset(c,0,sizeof c);
        for(i=1;i<=5;i++)
            b[i]=a[i];
        sort(b+1,b+6);
        sum=0;
        for(i=1;i<=5;i++)
        {
            sum+=b[i];
            c[b[i]]++;
        }
        int cd=code();
        q[cd]+=now;
        ck[cd]=1;
        for(i=1;i<=14;i++)
            g[cd][i]=calc(i);
        return;
    }
    int i;
    for(i=1;i<=6;i++)
    {
        a[x]=i;
        dfs(x+1,now*p[i]);
    }
}
inline void upmax(double &a,double b)
{
    if(b>a)
        a=b;
}
int main()
{
    int t;
    scanf("%d",&t);
    int i,j,k;
    double *e;
    double res;
    int tt=0;
    memset(ck,0,sizeof ck);
    while(t--)
    {
        for(i=1;i<=6;i++)
            scanf("%lf",&p[i]);
        memset(q,0,sizeof q);
        dfs(1,1);
        f[(1<<14)-1]=0;
        for(i=(1<<14)-2;i>=0;i--)
        {
            double ss=0;
            int cnt=0;
            for(j=1;j<=14;j++)
                if(!(i&(1<<(j-1))))
                {
                    d[++cnt]=j;
                    h[cnt]=f[i^(1<<(j-1))];
                }
            for(j=0;j<7776;j++)
            {
                if(!ck[j])
                    continue;
                e=g[j];
                res=0;
                for(k=1;k<=cnt;k++)
//                  res=max(res,h[k]+e[d[k]]);
                    upmax(res,h[k]+e[d[k]]);
                ss+=res*q[j];
            }
            f[i]=ss;
        }
        printf("Case #%d: %.6lf\n",++tt,f[0]);
    }
    return 0;
}