ZOJ1227-Free Candies

题目大意：给你4堆长度为n的糖果，每种糖果有一种颜色([1,20])；和一个篮子，你每次可以在每堆最上面的那个糖果中取走一个放在篮子里，如果篮子里有相同的，就放到结果里面，如果篮子里的糖果颜色都不一样，并且已经有5个了，就失败，问你最多可以取走多少对糖果？

题目解析：因为n才40，我们可以定义dp[41][41][41][41]表示每一堆取走多少个，这样时间复杂度为O(n^4)，不会超时，可以暴力dp，这里要用到DFS，深度优先搜索，并且要用到记忆化搜索，所以dfs的时候要注意恢复初始状态；

AC代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<memory.h>
using namespace std;

int n,dp[41][41][41][41],height[5],ans,number[41][4];
void dfs(int basket[],int num)
{
	int i,j;
	int tempnum,tempba[5];
	bool flag=0;
	tempnum=num;
	for(i=0;i<5;i++)
	{
		tempba[i]=basket[i];
		if(basket[i]==0)
		flag=1;
	}
	if(flag==0)
		return ;
	for(i=0;i<4;i++)
	{
		if(height[i]<n)
		{
			flag=0;
			for(j=0;j<5;j++)
			{
				if(basket[j]==number[height[i]][i])
				{
					flag=1;
					num++;
					basket[j]=0;
					break;
				}
			}
			if(flag==0)
			{
				for(j=0;j<5;j++)
				{
					if(basket[j]==0)
					{
						basket[j]=number[height[i]][i];
						break;
					}
				}
			}
			height[i]++; 
			if(dp[height[0]][height[1]][height[2]][height[3]]<num)
			{
				dp[height[0]][height[1]][height[2]][height[3]]=num;
				if(num>ans)
				ans=num;
				dfs(basket,num);
			}
			height[i]--;
			num=tempnum;
			memcpy(basket,tempba,sizeof(tempba));
		}
	}
}
int main()
{
	int basket[5],i,j;
	while(scanf("%d",&n)!=EOF)
	{
		for(i=0;i<n;i++)
			for(j=0;j<4;j++)
				scanf("%d",&number[i][j]);
		memset(dp,0xff,sizeof(dp));
		memset(basket,0,sizeof(basket));
		memset(height,0,sizeof(height));
		dp[0][0][0][0]=0;
		ans=0;
		dfs(basket,0);
		printf("%d\n",ans);
	}
	return 0;
}