UVA 10118 简单DP

题意

四堆糖果，有一个容量为5的篮子，每次可以从糖果堆顶端取一个糖果，若篮子中恰好有相同颜色的糖果，则将该糖果装入口袋中。问最多可以取多少个糖果。

题解

设dp[i][j][k][m]为四堆分别取了i,j,k,m个后，最多还能取多少个糖果。使用hash标记篮子中有的糖果颜色，DFS记忆化搜索即可得解。

代码

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

using namespace std;
int mp[50][5];
int dp[50][50][50][50];
int top[50];
int n;

int dfs(int count,bool hash[]){
    if(dp[top[0]][top[1]][top[2]][top[3]]!=-1){
        return dp[top[0]][top[1]][top[2]][top[3]];
    }
    if(count==5)
        return dp[top[0]][top[1]][top[2]][top[3]]=0;
    int ans=0;
    for(int i=0;i<4;i++){
        if(top[i]==n)
            continue;
        int color=mp[top[i]][i];
        top[i]+=1;
        if(hash[color]){
            hash[color]=false;
            ans=max(ans,dfs(count-1,hash)+1);
            hash[color]=true;
        }else{
            hash[color]=true;
            ans=max(ans,dfs(count+1,hash));
            hash[color]=false;
        }
        top[i]-=1;
    }
    return dp[top[0]][top[1]][top[2]][top[3]]=ans;
}

int main()
{
    while(scanf("%d",&n)!=EOF){
        if(n==0)
            break;
        for(int i=0;i<n;i++){
            for(int j=0;j<4;j++){
                scanf("%d",&mp[i][j]);
            }
        }
        memset(dp,-1,sizeof(dp));
        memset(top,0,sizeof(top));
        bool hash[30];
        memset(hash,false,sizeof(hash));
        printf("%d\n",dfs(0,hash));
    }
    return 0;
}