题目链接:https://ac.nowcoder.com/acm/problem/200190
题意:给n*m的矩阵,每次只能取一行或一列,取完后该行或该列变为0,总共能取k次,问怎么取才能使获得的数最大。
思路:n和m都很小,可以想到枚举选出k个行和列,但是最坏的时间复杂度为2^30次方会T,因此可以想到如果只枚举行或列选取哪些,另一边直接贪心的从大到小选取就好了,但是如何枚举呢,这边要采用二进制的方式,1表示选中,0表示未选中。
二进制枚举参考:https://blog.csdn.net/sugarbliss/article/details/81099340
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[20][20],b[20][20],temp[20];
bool cmp(int a,int b){
return a>b;
}
int main()
{
int n,m,k,s=0;
cin>>n>>m>>k;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
cin>>a[i][j];s+=a[i][j];b[i][j]=a[i][j];
}
}
if(k>=n||k>=m) cout<<s;
else{
int ans=0;
for(int i=0;i<(1<<n);i++){
s=0;vector<int> v;int num=0;
for(int j=0;j<n;j++) temp[j]=0;
for(int j=0;j<n;j++){
for(int kk=0;kk<m;kk++) b[j][kk]=a[j][kk];
}
for(int j=0;j<n;j++){
if((1<<j)&i){
temp[j]=1;num++;
}
else temp[j]=0;
}
if(num>k) continue;
for(int j=0;j<n;j++){
if(temp[j]){
for(int kk=0;kk<m;kk++) s+=b[j][kk],b[j][kk]=0;
}
}
for(int kk=0;kk<m;kk++){
int ss=0;
for(int j=0;j<n;j++) ss+=b[j][kk];
v.push_back(ss);
}
sort(v.begin(),v.end(),cmp);
for(int j=0;j<k-num;j++) s+=v[j];
ans=max(ans,s);
}
cout<<ans;
}
return 0;
}