题目大意是有一个二维矩阵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;
}