UVa 1508 Equipment 解题报告（枚举子集）

    UVa网页上看不到题目，点PDF就有。地址为：http://uva.onlinejudge.org/external/15/1508.pdf

    个人觉得非常好的一道题目。首先，k >= 5 的情况很好想，每一项取最大值即可。n==k的情况也可以这样。k==1的情况可以直接选择和最大的装备。

    剩下的就是k等于2,3,4的情况了，这些我也想不到。n有10000，就算k==2，暴力很可能会超时。

    参考了大牛们的做法：枚举子集。首先，最终的最大值是k件装备中每一项的最大值的和，那么对应的某间装备，最终结果的最大值里可能有第二项，第三项来自于这件装备。

    我们用二进制表示最终的最大值使用了某件装备的哪些项。比如 0 1 1 0 0 表示使用这件装备的第二项和第三项。初始化时枚举每件装备所有的情况，记录最大值。

    最终我们需要求出集合1 1 1 1 1的最大值，我们可以枚举它的子集，比如1 0 1 0 0 和 0 1 0 1 1。最多有k个子集，更新出最大值即可。

    因为从枚举n无法完成，从而从每一项的方向入手。很好的思维。代码如下：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <set>
#include <map>
#include <vector>
#include <queue>
#include <string>
#include <iostream>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
#define ff(i, n) for(int i=0;i<(n);i++)
#define fff(i, n, m) for(int i=(n);i<=(m);i++)
#define dff(i, n, m) for(int i=(n);i>=(m);i--)
void work();
int main()
{
#ifdef ACM
    freopen("in.txt", "r", stdin);
#endif // ACM

    work();
}

/****************************************/

int ma[1<<5];

int dp(int st, int k)
{
    if(k == 1)
        return ma[st];

    int ans = 0;
    ff(s, 1<<5) if((st | (st^s)) <= st)
        ans = max(ans, ma[s] + dp(st^s, k-1));
    return ans;
}

void work()
{
    int T;
    scanf("%d", &T);
    fff(cas, 1, T)
    {
        int n, k;
        scanf("%d%d", &n, &k);

        int ans = 0;
        int tmp;
        int num[5];

        if(k >= 5 || n == k)
        {
            memset(num, 0, sizeof(num));
            ff(i, n) ff(j, 5)
                scanf("%d", &tmp), num[j] = max(num[j], tmp);
            ff(j, 5) ans += num[j];
        }
        else
        {
            memset(ma, 0, sizeof(ma));

            ff(i, n)
            {
                ff(j, 5)
                    scanf("%d", num + j);

                ff(s, 1<<5)
                {
                    tmp = 0;
                    ff(j, 5) if(s&(1<<j))
                        tmp += num[j];
                    ma[s] = max(ma[s], tmp);
                }
            }
            ans = dp((1<<5)-1, k);
        }

        printf("%d\n", ans);
    }
}