题目大意:
有4堆糖果,每一堆有n个糖果,有一个篮子,这个篮子的最大容量为5块糖,每一次从这四堆中选择一堆,然后将最上面的糖果拿下来,放到篮子里面去,如果有相同两块糖的就把这;两块糖拿出来,如果达到了5块糖,并且这5块糖是两两不相同的就结束游戏,问你,最多能拿出多少对糖
动态规划,状态为篮子里的糖果数,用k1 k2 k3 k4代表每堆糖的高度,即4维矩阵进行存储,代表此时糖堆的状态时所能得到的糖的最大数目,然后每次取糖,判断是否有一样的,如果有一样的糖就放入口袋,没有则放入篮子。每个糖堆状态所得到的糖的最大数目都应该等于
Code:
- #include<iostream>
- #include<cstdio>
- #include<cstring>
- using namespace std;
- const int maxn = 45;
- int num[maxn][4]; //存放输入的数字
- int top[4]; //现在每一堆拿到第几个了
- int dp[maxn][maxn][maxn][maxn]; //递推,记录结果的数组
- int vis[4*maxn]; //记录篮子里有没有现在这个数字
- int n;
- int DP(int k)
- {
- int & ans = dp[top[0]][top[1]][top[2]][top[3]];
- if(ans != -1) return ans;
- ans = 0;
- if(k >= 5) return ans; //篮子里面最多有5个,如果有5个不一样的就退出
- for(int i = 0; i < 4; i++){
- if(top[i] == n) continue; //这一堆已经全部都拿完了
- ++top[i];
- if(vis[num[top[i]][i]]){ //在篮子里面有一样的
- vis[num[top[i]][i]] = false; //把一样的拿出去
- ans = max(ans, DP(k-1)+1); // !!!!!
- vis[num[top[i]][i]] = true; //恢复原样
- }
- else{
- vis[num[top[i]][i]] = true;
- ans = max(ans, DP(k+1)); // !!!!
- vis[num[top[i]][i]] = false;
- }
- --top[i];
- }
- return ans;
- }
- int main()
- {
- while(scanf("%d", &n) && n){
- for(int i = 1; i <= n; i++){
- for(int j = 0; j < 4; j++)
- scanf("%d", &num[i][j]);
- }
- memset(top, 0, sizeof(top));
- memset(dp, -1, sizeof(dp));
- memset(vis, false, sizeof(vis));
- printf("%d\n", DP(0));
- }
- return 0;
- }