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;
}