UVA - 10118 Free Candies

来源:互联网 发布:单片机使用教程 编辑:程序博客网 时间:2024/05/29 06:48
题意:有4堆糖果,有一个篮子最多只能放五个,从这些糖果任意拿一个放进篮子里,如果篮子里有两个一样的就放进口袋里,求最多放进口袋几种,因为数据不是很大,所以直接用4维的表示,dp[x1][x2][x3][x4]表示当4堆还剩多少的时候最多放进口袋几种,用二进制表示是否在篮子里
#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int MAXN = 42;int dp[MAXN][MAXN][MAXN][MAXN];  //当每堆剩下多少时口袋的数量int n,arr[5],a[5][MAXN];int dfs(int basket,int candy){    if (dp[arr[1]][arr[2]][arr[3]][arr[4]] != -1)        return dp[arr[1]][arr[2]][arr[3]][arr[4]];    else {        int sum,t;        int ans = 0;        for (int i = 1; i <= 4; i++){            sum = 0;            ++arr[i];            if (arr[i] <= n){                if ((t = 1<<a[i][arr[i]]) & basket)  //判断是否在篮子里                    sum = dfs((~t)&basket,candy-1) + 1;                else if (candy < 4)                    sum = dfs(t|basket,candy+1);  //将这颗糖加进篮子            }            --arr[i];   //回溯            ans = max(ans,sum);        }               return dp[arr[1]][arr[2]][arr[3]][arr[4]] = ans;         }}int main(){    while (scanf("%d",&n) != EOF && n){        for (int i = 1; i <= n; i++)            for (int j = 1; j <= 4; j++)                scanf("%d",&a[j][i]);        memset(dp,-1,sizeof(dp));        memset(arr,0,sizeof(arr));        printf("%d\n",dfs(0,0));    }    return 0;}



原创粉丝点击