UVA1637Double Patience(概率 + 记忆化搜索)

训练指南P327

题意：36张牌分成9堆， 每堆4张牌。每次拿走某两堆顶部的牌，但需要点数相同。如果出现多种拿法则等概率的随机拿。 如果最后拿完所有的牌则游戏成功，求成功的概率。

开个9维数组表示每一堆的状态，模拟搜索一下

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int Max = 6;
char s[Max][Max];
int A[10][Max * Max];
int vis[Max][Max][Max][Max][Max][Max][Max][Max][Max];
double d[Max][Max][Max][Max][Max][Max][Max][Max][Max];
double dfs(int s1, int s2, int s3, int s4, int s5, int s6, int s7, int s8, int s9)
{
if (vis[s1][s2][s3][s4][s5][s6][s7][s8][s9])
return d[s1][s2][s3][s4][s5][s6][s7][s8][s9];
vis[s1][s2][s3][s4][s5][s6][s7][s8][s9] = 1;
int T[10] = {0, s1, s2, s3, s4, s5, s6, s7, s8, s9};
bool flag = true;
for (int i = 1; i <= 9; i++)
{
if (T[i])
{
flag = false;
break;
}
}//如果全是0，表示结束，这个状态为1
if (flag)
{
return d[s1][s2][s3][s4][s5][s6][s7][s8][s9] = 1.0;
}
int tot = 0;
double cnt = 0;
for (int i = 1; i <= 9; i++)
{
for (int j = i + 1; j <= 9; j++)
{
if (T[i] > 0 && T[j] > 0 && A[i][ T[i] ] == A[j][ T[j] ])
{
T[i]--;
T[j]--;
tot++; // 当前有几种选择方案
cnt += dfs(T[1], T[2], T[3], T[4], T[5], T[6], T[7], T[8], T[9]); // 选择 i 堆 和 j 堆之后剩下的全部取光的概率
T[i]++;
T[j]++;
}
}
}
if (tot > 0)
d[s1][s2][s3][s4][s5][s6][s7][s8][s9] = cnt / (1.0 * tot); //  (1 / tot ) * cnt;当前状态概率
return d[s1][s2][s3][s4][s5][s6][s7][s8][s9];
}
int main()
{
while (scanf("%s%s%s%s", s[1], s[2], s[3], s[4]) != EOF)
{
for (int i = 1; i <= 4; i++)
A[1][i] = s[i][0] - '0';  // A[i][j]用于记录第i堆第j个
for (int i = 2; i <= 9; i++)
{
for (int j = 1; j <= 4; j++)
{
scanf("%s", s[j]);
A[i][j] = s[j][0] - '0';
}
}
memset(vis, 0, sizeof(vis));
memset(d, 0, sizeof(d));
dfs(4, 4, 4, 4, 4, 4, 4, 4, 4);  // 所有的堆都是4个开始搜索
printf("%.6lf\n", d[4][4][4][4][4][4][4][4][4]);
}
return 0;
}

View Code