UVA 10118 Free Candies 记忆化搜索 dp紫书9-2

UVA 10118

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>

using namespace std;
const int maxn=50;
int a[4][maxn],n;
int d[maxn][maxn][maxn][maxn];///d[i][j][k][l]四堆分别选择了i j k l个糖果时口袋里的糖果
int top[4];
int dp(int cnt,bool ok[])///篮子里的糖果数为cnt ok数组为篮子里有的糖果
{
    if(d[top[0]][top[1]][top[2]][top[3]]!=-1) 
        return d[top[0]][top[1]][top[2]][top[3]];
    if(cnt==5) return 0;
    int ans=0;
    for(int i = 0; i < 4; ++i) {
        if(top[i]==n) continue;
        int color=a[i][top[i]];
        top[i]++;
        if(ok[color]) {
            ok[color]=false;
            ans=max(ans,dp(cnt-1,ok)+1);
            ok[color]=true;
        }
        else {
            ok[color]=true;
            ans=max(ans,dp(cnt+1,ok));
            ok[color]=false;
        }
        top[i]-=1;
    }
    return d[top[0]][top[1]][top[2]][top[3]]=ans;
}
int main()
{
    while(cin>>n&&n) {
        for(int i = 0; i < n; ++i) {
            for(int j = 0; j < 4; ++j) {
                cin>>a[j][i];
            }
        }
        bool ok[25]={0};
        memset(d,-1,sizeof d);
        top[0]=top[1]=top[2]=top[3]=0;
        cout<<dp(0,ok)<<endl;;
    }
    return 0;
}