矩阵消除游戏(二进制枚举+贪心)

题目链接: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;
} 

 

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值