码迷,mamicode.com
首页 > 其他好文 > 详细

uva 10118 Free Candies

时间:2018-05-06 00:23:58      阅读:196      评论:0      收藏:0      [点我收藏+]

标签:int   \n   memset   amp   code   col   表示   dfs   sizeof   

题意:

有4堆糖,每堆有n个,每次从某一堆的堆顶拿一个放进篮子里,如果篮子里有2个颜色相同的糖果,那么就可以放进袋子里。

当篮子里有5个糖果并且没有相同颜色的糖果时,这个时候就不能再拿了。

问最多可以拿多少对颜色相同的糖果。

思路:

记忆化搜索。

设dp[x][y][z][w]表示第一堆拿走了x个,第二堆拿走了y个,第三堆拿走了z,第四堆拿走了w个时,可以取得的最大对数。

篮子里的糖果,因为最多只有20种颜色,所以可以状态压缩表示,当大于等于5个的时候,就无法再继续了。

代码:

 1 #include <stdio.h>
 2 #include <string.h>
 3 #include <algorithm>
 4 using namespace std;
 5 const int N = 20;
 6 int n;
 7 int a[45][5];
 8 int dp[45][45][45][45];
 9 int dfs(int x,int y,int z,int w,int sta)
10 {
11     int &ans = dp[x][y][z][w];
12     if (~ans) return ans;
13     int cnt = 0;
14     for (int i = 0;i < 20;i++)
15     {
16         if (sta & (1<<i)) cnt++;
17     }
18     if (cnt >= 5) return ans = 0;
19     if (x < n)
20     {
21         int tmp;
22         if (sta & (1 << a[x+1][1]-1))
23         {
24             tmp = dfs(x+1,y,z,w,sta ^ (1 << a[x+1][1]-1)) + 1;
25         }
26         else
27         {
28             //printf("hh\n");
29             tmp = dfs(x+1,y,z,w,sta ^ (1 << a[x+1][1]-1));
30         }
31         ans = max(ans,tmp);
32     }
33     if (y < n)
34     {
35         int tmp;
36         if (sta & (1 << a[y+1][2]-1))
37         {
38             tmp = dfs(x,y+1,z,w,sta ^ (1 << a[y+1][2]-1)) + 1;
39         }
40         else
41         {
42             tmp = dfs(x,y+1,z,w,sta ^ (1 << a[y+1][2]-1));
43         }
44         ans = max(ans,tmp);
45     }
46     if (z < n)
47     {
48         int tmp;
49         if (sta & (1 << a[z+1][3]-1))
50         {
51             tmp = dfs(x,y,z+1,w,sta ^ (1 << a[z+1][3]-1)) + 1;
52         }
53         else
54         {
55             tmp = dfs(x,y,z+1,w,sta ^ (1 << a[z+1][3]-1));
56         }
57         ans = max(tmp,ans);
58     }
59     if (w < n)
60     {
61         int tmp;
62         if (sta & (1 << a[w+1][4]-1))
63         {
64             tmp = dfs(x,y,z,w+1,sta ^ (1 << a[w+1][4]-1)) + 1;
65         }
66         else
67         {
68             tmp = dfs(x,y,z,w+1,sta ^ (1 << a[w+1][4]-1));
69         }
70         ans = max(ans,tmp);
71     }
72     if (ans == -1) return ans = 0;
73     
74     return ans;
75 }
76 int main()
77 {
78     while (scanf("%d",&n) != EOF && n)
79     {
80         memset(dp,-1,sizeof(dp));
81         for (int i = 1;i <= n;i++)
82         {
83             for (int j = 1;j <= 4;j++) scanf("%d",&a[i][j]);
84         }
85         int ans = dfs(0,0,0,0,0);
86         printf("%d\n",ans);
87     }
88     return 0;
89 }
90 /*
91 3 1 2 3 4 5 6 7 8 1 2 3 4
92 */

 

uva 10118 Free Candies

标签:int   \n   memset   amp   code   col   表示   dfs   sizeof   

原文地址:https://www.cnblogs.com/kickit/p/8996717.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!