题目:给你一副扑克牌,54张,让你求出取出黑梅红方最少分别为a,b,c,d张 的取的次数的期望。其中大小王可以当做四种牌的任意一种。
题解:dp[a][b][c][d][p][q]表示每种牌分别有a,b,c,d张,Joker的状态分别作为p和q出现时的期望值,然后记忆化搜索。(dp状态想到了,但是没能实现,参考一位大佬的方法%%%%%)
#include<bits/stdc++.h>
using namespace std;
#define inf 1e9
double dp[14][14][14][14][5][5];
int num[5];
int check(int a,int b,int c,int d,int p,int q)//判断是不是各种牌都满足要取的数量了
{
int x[5]={0};
x[1]=a;x[2]=b;x[3]=c;x[4]=d;
x[p]++;x[q]++;
for(int i=1;i<=4;i++)
if(x[i]<num[i])
return 0;
return 1;
}
double dfs(int a,int b,int c,int d,int p,int q)
{
if(check(a,b,c,d,p,q)) return dp[a][b][c][d][p][q]=0;
if(dp[a][b][c][d][p][q]>=0) return dp[a][b][c][d][p][q];
int sum=54-a-b-c-d;
if(p) sum--;
if(q) sum--;
if(sum==0) return 0;
double ans=0.0;
if(a<13) ans+=dfs(a+1,b,c,d,p,q)*(13-a)/sum;//下一张取a种牌的期望
if(b<13) ans+=dfs(a,b+1,c,d,p,q)*(13-b)/sum;
if(c<13) ans+=dfs(a,b,c+1,d,p,q)*(13-c)/sum;
if(d<13) ans+=dfs(a,b,c,d+1,p,q)*(13-d)/sum;
if(p==0)
{
double tmp=inf;
for(int i=1;i<=4;i++)
tmp=min(tmp,dfs(a,b,c,d,i,q));//因为大小王是人为规定的当哪张牌使,所以取最小的
ans+=tmp/sum;
}
if(q==0)
{
double tmp=inf;
for(int i=1;i<=4;i++)
tmp=min(tmp,dfs(a,b,c,d,p,i));
ans+=tmp/sum;
}
ans+=1.0;//本次取一张牌
return dp[a][b][c][d][p][q]=ans;
}
int t,cas=0;
int main()
{
scanf("%d",&t);
while(t--)
{
for(int i=1;i<=4;i++)
scanf("%d",&num[i]);
int delt=0;
for(int i=1;i<=4;i++)
delt+=max(num[i]-13,0);//判断让取的牌是否合理
if(delt>2)
{
printf("Case %d: -1\n",++cas);
continue;
}
memset(dp,-1,sizeof dp);
double ans=dfs(0,0,0,0,0,0);
printf("Case %d: %.8lf\n",++cas,ans);
}
return 0;
}