UVa 10118 Free Candies 记忆化搜索

题目大意：有4堆糖果，每堆有n个，有一只最多能容5个糖果的篮子。现在，要把糖果放到篮子里，如果篮子中有相同颜色的糖果，放的人就可以拿到自己的口袋。如果放的人足够聪明，问他最多能得到多少对糖果。

题目分析：很显然的多阶段决策。定义dp(a,b,c,d)为每堆糖果分别拿掉a、b、c、d块之后最多能获得得糖果对数。则决策有4个，以第一堆为例，状态转移方程为：dp(a,b,c,d)=dp(a+1,b,c,d) （如果拿掉第一堆的第a+1个不会产生相同颜色）、dp(a,b,c,d)=dp(a+1,b,c,d)+1 （如果拿掉第一堆的第a+1个会产生相同颜色）。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
using namespace std;

int n;
int dp[41][41][41][41], M[41][4];
bool visit[25];
int top[4];
int DP(int count)
{
int &ans = dp[top[0]][top[1]][top[2]][top[3]];
if(ans != -1) return ans;
ans = 0;
if(count >= 5) return ans;
for(int i = 0; i < 4; i++)
{
if(top[i] == n) continue;
top[i]++;
if(visit[M[top[i]][i]]) {
visit[M[top[i]][i]] = false;
ans = max(ans, DP(count-1)+1);
visit[M[top[i]][i]] = true;
}
else {
visit[M[top[i]][i]] = true;
ans = max(ans, DP(count+1));
visit[M[top[i]][i]] = false;
}
top[i]--;
}
return ans;
}

int main()
{
freopen("0.0.txt", "r", stdin);
while(scanf("%d", &n) && n)
{
for(int i = 1; i <= n; i++)
{
for(int j = 0; j < 4; j++)
{
scanf("%d", &M[i][j]);
}
}
memset(visit, false, sizeof(visit));
memset(dp, -1, sizeof(dp));
memset(top, 0, sizeof(top));
int result = DP(0);
printf("%d\n", result);
}

return 0;
}