牛客网 Nowcoder Girl挑战赛 Problem 4 勇敢的妞妞

题目大意是有一个二维矩阵r[i][j],0<i<N,0<j<5，从中选K行使得sum_{i \in K}max_{j}(r[i][j])最大。i是武器，j是属性。

首先，如果K>=5，那么先针对每一个属性，选择该属性值j最大的武器i，剩下的武器随便选即可。

如果K<5，因为属性个数<=5，对于每一个属性子集 J\in {0,1,2,3,4}，都存在一个武器i，使得i的子集中属性值之和

最大，i.e.，\sum_{j\in J} r[i][j] is maximum. 因为一共有2^5-1=32个子集，这样的武器一共有32个。

那么，最终的选择一定是在这32个武器中选。假设选了武器a和b，a贡献了属性0，1，2（a的0,1,2属性值比b的大），b贡献了属性3，4。如果a不是这32个武器中的，那么存在另一个武器c，且c的0,1,2属性和比a大，那么结果会更忧。

#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<string>
#include<cmath>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<cstring>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<windows.h>
using namespace std;

const int maxn=10010;
int K;
int N;
int r[maxn][5];
int max_prop[32];//save items with max value in some properties
int ans=0;
//int cnt=0;
vector<int>candidate=vector<int>();
set<int>save=set<int>();
void dfs(int st,int sum[5],int num)
{
    if(num==K)
    {
        int tmp=0;
        for(int i=0;i<5;i++)
        {
            tmp+=sum[i];
        }
        ans=max(ans,tmp);
//        cnt++;
//        cout<<cnt<<endl;
        return;
    }
    for(int i=st;i<candidate.size();i++)
    {
        int tmpsum[5];
        memset(tmpsum,0,sizeof(tmpsum));
        for(int j=0;j<5;j++)
        {
            tmpsum[j]=max(sum[j],r[candidate[i]][j]);
        }
//        cout<<"prev "<<i<<" next "<<i+1<<endl;
        dfs(i+1,tmpsum,num+1);
    }
    return;
}
int main()
{
    freopen("input.txt","r",stdin);
    while(cin>>N>>K)
    {
        memset(r,0,sizeof(r));
        memset(max_prop,0,sizeof(max_prop));
        ans=0;
        candidate.clear();
        save.clear();
        for(int i=0;i<N;i++)
        {
            for(int j=0;j<5;j++)
            {
                cin>>r[i][j];
            }
        }
        if(K>=5)
        {
            for(int j=0;j<5;j++)
            {
                int tmp=0;
                for(int i=0;i<N;i++)
                {
                    if(tmp<r[i][j])
                    {
                        tmp=r[i][j];
                    }
                }
                ans+=tmp;
            }
            cout<<ans<<endl;
            continue;
        }
        for(int x=1;x<(1<<5);x++)
        {
            int property[5];
            memset(property,0,sizeof(property));
            for(int i=0;i<5;i++)
            {
                if((x&(1<<i))!=0)
                {
                    property[i]=1;
                }
            }
            int maxid=-1;
            int maxpty=0;
            for(int i=0;i<N;i++)
            {
                int tmp=0;
                for(int j=0;j<5;j++)
                {
                    if(property[j]==1)
                    {
                        tmp+=r[i][j];
                    }
                }
                if(tmp>maxpty)
                {
                    maxid=i;
                    maxpty=tmp;
                }
            }
            max_prop[x]=maxid;
            if(save.find(maxid)==save.end())
            {
                save.insert(maxid);
                candidate.push_back(maxid);
            }

//            cout<<x<<" "<<maxid<<endl;
        }
//        for(int i=0;i<candidate.size();i++)
//        {
//            cout<<candidate[i]<<" ";
//        }
//        cout<<endl;
        int sum[5];
        memset(sum,0,sizeof(sum));
        dfs(0,sum,0);
        cout<<ans<<endl;

    }
    return 0;
}