hdu 1514 Free Candies 解题报告

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1514

题目意思：有4堆糖果，每堆糖果有n个，从上到下排好，取糖果只能从上往下取，取完的糖果放在篮子里，篮子里最多放5个，如果篮子里有两个颜色相同的糖果则可以取走放进口袋里，问最多能取走多少对糖果放进口袋。n<=40, 糖果颜色最多20种。

     这题是抄这个人滴：http://fudq.blog.163.com/blog/static/1913502382014239225290/

     有些地方看得不太懂，本来想搜状态压缩DP来体会下的，阴差阳错、迷迷糊糊找到了这题= =.......可能水平不太够啦，先留着吧.......能理解到的我都写注释了，不能理解的打了个“？”

     只能说，DP好抽象啊，泪 >_< ，慢慢来吧，继续努力！！！

    

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
using namespace std;

const int candy_num = 40 + 5;
const int basket_num = 5 + 2;

// dp[i][j][k][l]: 四个糖堆分别拿走了几个
//  第1个pile被取走i个，第2个pile被取走j个，第3个pile被取走k个，第4个pile被取走l个时pocket能装入的candy数
// h[i]：记录第i个糖堆被取走了多少糖果
int dp[candy_num][candy_num][candy_num][candy_num];
int h[basket_num], f[candy_num][basket_num];
int ans, n;

set<int> S;
set<int> ::iterator it;

int check(int a)
{
    it = S.find(a);
    if (it != S.end())   // basket里面有颜色跟当前颜色(a)相同的糖果，从basket里拿出
    {
        S.erase(a);
        return 1;
    }
    S.insert(a);   // basket中没有a这种颜色，放入basket中
    return 0;
}

int dfs(int tmp)
{
    if (dp[h[0]][h[1]][h[2]][h[3]] != -1)   // 这个状态搜过
        return dp[h[0]][h[1]][h[2]][h[3]];
    int tt = 0;
    for (int i = 0; i < 4; i++)   // 从四个basket中拿糖果
    {
        if (h[i] < n && S.size() < 5)
        {
            int t = check(f[h[i]][i]);
            if (S.size() < 5)
            {
                h[i]++;   // 第i个pile里被取走一个candy
                tt = max(tt, dfs(tmp+t));   // ?
                h[i]--;
            }
            tt = max(tt, tmp+t);     // ?
            if (t == 1)               
                S.insert(f[h[i]][i]);
            else
                S.erase(f[h[i]][i]);
        }
    }
    return dp[h[0]][h[1]][h[2]][h[3]] = tt;
}

int main()
{
    while (scanf("%d", &n) != EOF && n)
    {
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < 4; j++)
                scanf("%d", &f[i][j]);
        }
/*        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < 4; j++)
                printf("f[%d][%d] = %d
", i, j, f[i][j]);
        }
*/
        ans = 0;
        h[0] = h[1] = h[2] = h[3] = 0;
        memset(dp, -1, sizeof(dp));
        printf("%d
", dfs(0));
    //    printf("dp[%d][%d][%d][%d] = %d
", h[0], h[1], h[2], h[3], dp[h[0]][h[1]][h[2]][h[3]]);
    }
    return 0;
}