题意:小明要拿糖果,糖果有4叠,每次只能从任意一叠糖果的顶部拿一颗糖放到篮子里,如果篮子里的糖果出现颜色一样的就拿出这对糖果,篮子里的糖果最多5个,问最多可以拿出多少对颜色相同的糖果。
题解:用f[i][j][k][l]存4叠糖果拿了i,j,k,l个时最多能拿出几对糖果。回溯,中间可以通过f数组减少时间。
#include <stdio.h>
#include <string.h>
const int N = 45;
int n, f[N][N][N][N];
int s[N][4];
int st[4], vis[N];
int dp(int cnt) {
int &temp = f[st[0]][st[1]][st[2]][st[3]];
if (temp != -1)
return temp;
if (cnt == 5)
return temp = 0;
temp = 0;
for (int i = 0; i < 4; i++) {
if (st[i] == n)
continue;
if (vis[s[st[i]][i]] == 1) {
vis[s[st[i]][i]] = 0;
st[i]++;
int a = dp(cnt - 1) + 1;
temp = temp > a ? temp : a;
vis[s[st[i] - 1][i]] = 1;
}
else {
vis[s[st[i]][i]] = 1;
st[i]++;
int a = dp(cnt + 1);
temp = temp > a ? temp : a;
vis[s[st[i] - 1][i]] = 0;
}
st[i]--;
}
return temp;
}
int main() {
while (scanf("%d", &n) && n) {
memset(f, -1, sizeof(f));
memset(vis, 0, sizeof(vis));
for (int i = 0; i < n; i++)
for (int j = 0; j < 4; j++)
scanf("%d", &s[i][j]);
st[0] = st[1] = st[2] = st[3] = 0;
printf("%d\n", dp(0));
}
return 0;
}