一遍AC,0.132s
说说我的思路吧,首先考虑如何表示状态,不难发现,情况非常多,因为怎么拿都行,所以只好增加维度,开四维数组。
由于结构比较无序,所以选择了记忆化搜索 ,受前面《校长的烦恼》的启发,即使我们还要维护篮子中糖果情况,但是我们只需开四维数组就已经足够表示所有的状态了 ,我们大可以将篮子中的情况以及口篮子中糖果的数量放在函数的参数中来维护就可以了,那么我们不难用d[a][b][c][d]表示各堆糖果取到当前状态时口袋中糖果的最大对数 。
如何维护当前篮子中糖果的颜色情况呢? 我看别人题解中方法很多,我认为比较简单的还是我这个方法,只需要用一个整数通过二进制运算表示成一个集合的方法,对,因为当前篮子中糖果的颜色是不同的(相同的立刻会拿到口袋里) ,因此就可以这样简单的表示了 。
细节参加代码:
#include<bits/stdc++.h>
using namespace std;
int n,f[42][42][42][42],col[42][6];
int dp(int a,int b,int c,int d,int s,int cnt) {
if(cnt == 5) return 0;//边界条件
int& ans = f[a][b][c][d];//对于状态较多的时候这样通常会很方便
if(ans >= 0) return ans;
int s0;//集合s维护当前篮子里的糖果情况,由于存在于篮子中的糖果颜色都是唯一的,所以可以用集合表示
ans = 0;
if(a <= n) {
int v = col[a][1];
if(s & (1<<v)) {//该颜色在集合中已经存在
s0 = s ^ (1<<v);//移除
ans = max(ans,dp(a+1,b,c,d,s0,cnt-1)+1);
}
else {
s0 = s | (1<<v);//加入
ans = max(ans,dp(a+1,b,c,d,s0,cnt+1));
}
} //下同
if(b <= n) {
int v = col[b][2];
if(s & (1<<v)) {
s0 = s ^ (1<<v);
ans = max(ans,dp(a,b+1,c,d,s0,cnt-1)+1);
}
else {
s0 = s | (1<<v);
ans = max(ans,dp(a,b+1,c,d,s0,cnt+1));
}
}
if(c <= n) {
int v = col[c][3];
if(s & (1<<v)) {
s0 = s ^ (1<<v);
ans = max(ans,dp(a,b,c+1,d,s0,cnt-1)+1);
}
else {
s0 = s | (1<<v);
ans = max(ans,dp(a,b,c+1,d,s0,cnt+1));
}
}
if(d <= n) {
int v = col[d][4];
if(s & (1<<v)) {
s0 = s ^ (1<<v);
ans = max(ans,dp(a,b,c,d+1,s0,cnt-1)+1);
}
else {
s0 = s | (1<<v);
ans = max(ans,dp(a,b,c,d+1,s0,cnt+1));
}
}
return ans;
}
int main() {
while(~scanf("%d",&n)&&n) {
memset(f,-1,sizeof(f));
for(int i=1;i<=n;i++)
for(int j=1;j<=4;j++) scanf("%d",&col[i][j]);
printf("%d\n",dp(1,1,1,1,0,0));
}
return 0;
}