UVa 10118 记忆化搜索 Free Candies

假设在当前状态我们第i堆糖果分别取了cnt[i]个，那么篮子里以及口袋里糖果的个数都是可以确定下来的。

所以就可以使用记忆化搜索。

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

 const int maxn = ;
 const int maxm = ;

 int n;

 int candy[maxn][maxm];
 int d[maxn][maxn][maxn][maxn];

 struct State
 {
     int tot;        //篮子里糖果的数量
     int cnt[];     //第i堆糖果取了cnt[i]个
     int kind[];   //篮子里第i中糖果有ind[i]个
 };

 int dp(State s)
 {
     int& ans = d[s.cnt[]][s.cnt[]][s.cnt[]][s.cnt[]];
     if(ans >= ) return ans;
     if(s.tot == ) return ans = ;

     ans = ;
     for(int i = ; i < ; i++)
     {
         if(s.cnt[i] == n) continue;
         int color = candy[s.cnt[i]][i];
         s.cnt[i]++;
         if(s.kind[color])
         {
             s.tot--;
             s.kind[color] = ;
             ans = max(ans, dp(s) + );
             s.tot++;
             s.kind[color] = ;

         }
         else
         {
             s.tot++;
             s.kind[color] = ;
             ans = max(ans, dp(s));
             s.tot--;
             s.kind[color] = ;
         }
         s.cnt[i]--;
     }
     return ans;
 }

 int main()
 {
     while(scanf("%d", &n) ==  && n)
     {
         for(int i = ; i < n; i++)
             for(int j = ; j < ; j++) scanf("%d", &candy[i][j]);

         State s;
         s.tot = ;
         for(int i = ; i < ; i++) s.cnt[i] = ;
         for(int i = ; i <= ; i++) s.kind[i] = ;
         memset(d, -, sizeof(d));
         printf("%d\n", dp(s));
     }

     return ;
 }

代码君