矩阵消除游戏 牛客

矩阵消除游戏
行二进制枚举，列贪心处理

#include<bits/stdc++.h>
using namespace std;
#define pi pair<int,int>
#define int long long 
const int maxx=2e5+10;
int n,m,a[20][20],s1[20],b[20],ans,sh[20];
signed main(){
	int k;
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++) cin>>a[i][j],sh[i]+=a[i][j];
	}
	if(k>n) k=n;
	if(k>m) k=m;
	for(int i=0;i<(1<<n);i++){
		memset(b,0,sizeof(b));
		int cnt=0,sum=0;
		for(int j=0;j<n;j++){
			if(i&(1<<j)){
				cnt++;
				b[j+1]=1;
			}
		}
		int o=k-cnt;
		if(o>m||o<0) continue;
		for(int i=1;i<=n;i++){
			if(b[i]) sum+=sh[i];
		}
		memset(s1,0,sizeof(s1));
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				if(!b[i]) s1[j]+=a[i][j];
			}
		}
		sort(s1+1,s1+m+1);
		for(int j=m;o>0;j--,o--) sum+=s1[j];
		ans=max(ans,sum);
	}
	cout<<ans;	
	return 0;
}