题目大意:有4堆糖果,每堆有n个,有一只最多能容5个糖果的篮子。现在,要把糖果放到篮子里,如果篮子中有相同颜色的糖果,放的人就可以拿到自己的口袋。如果放的人足够聪明,问他最多能得到多少对糖果。
题目分析:很显然的多阶段决策。定义dp(a,b,c,d)为每堆糖果分别拿掉a、b、c、d块之后最多能获得得糖果对数。则决策有4个,以第一堆为例,状态转移方程为:dp(a,b,c,d)=dp(a+1,b,c,d) (如果拿掉第一堆的第a+1个不会产生相同颜色)、dp(a,b,c,d)=dp(a+1,b,c,d)+1 (如果拿掉第一堆的第a+1个会产生相同颜色)。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <string>
#include <string.h>
#include <cmath>
#include <sstream>
using namespace std;
const int maxn = 50;
int dp[maxn][maxn][maxn][maxn];
int pile[maxn][4], p[6];
int n;
int dfs(int num, int status)
{
int &ans = dp[p[0]][p[1]][p[2]][p[3]];
//修改dp数组时,同时修改ans。
if (ans != -1)
return ans;
if (num == 5)
return 0;
ans = 0;
for (int i = 0; i < 4; i++)
{
if (p[i] < n)
{
int st = 1<<pile[p[i]][i];
p[i]++;
if (status&st)//用与运算表示有相同颜色的糖果
ans = max(ans, dfs(num - 1, status^st) + 1);
else
ans = max(ans, dfs(num + 1, status^st));//异或表示放入取出一个糖果后的状态
p[i]--;
}
}
return ans;
}
int main()
{
while (cin >> n&&n!=0)
{
for (int i = 0; i < n; i++)
{
for (int j = 0; j < 4; j++)
{
cin >> pile[i][j];
}
}
memset(dp, -1, sizeof(dp));
memset(p, 0, sizeof(p));
cout << dfs(0, 0) << endl;;
}
}
在运算篮子里有没有相同种类糖果时,运用异或跟与的特性:
预处理:int st = 1<<pile[p[i]][i];
statued:1^2,st=3;
statue:1^2&3==false;
statued :1^2,st=3;
statue:1^2&3==true;但是按照提议要求,篮子里有不同糖果应该为false;