牛客练习赛58 C题

矩阵消除游戏

题目描述

牛妹在玩一个名为矩阵消除的游戏，矩阵的大小是n行m列，第i行第j列的单元格的权值为a i,j
，牛妹可以进行k个回合的游戏，在每个回合，牛妹可以选择一行或者选择一列，然后将这一行或者这一列的所有单元格中的权值变为0，同时牛妹的分数会加上这一行或者这一列中的所有单元格的权值的和。

牛妹想最大化她的得分，球球你帮帮她吧！

输入描述:

第一行三个整数n,m,k
接下来n行每行m个整数表示矩阵中各个单元格的权值。

输出描述:

输出一个整数表示牛妹能获得的最大分数。

---

二进制枚举行，根据把枚举所到的行进行清0，然后把列值重新计算排序，取前面所需的列值就行；

注意枚举行时不是1到2^k, 是1到 2 ^n，再进行判断1的个数是否大于k；

代码：

#include<bits/stdc++.h>
#define ll long long
#define pa pair<int,int>
#define lson k<<1
#define rson k<<1|1
#define inf 0x3f3f3f3f
//ios::sync_with_stdio(false);
using namespace std;
const int N=100100;
const int M=1000100;
const ll mod=1e9+7;
ll a[20][20];
ll r[20],l[20];
ll sum;
int n,m,k;
ll solve(int p){
	ll b[20][20];
	memcpy(b,a,sizeof(a));
	vector<int>ve;
	int tot=0;
	ll ans=0;
	while(p){
		tot++;
		if(p%2) ve.push_back(tot);
		p/=2;
	}
	if(ve.size()>k) return 0;
	for(int i=0;i<ve.size();i++){
		ans+=r[ve[i]];
		for(int j=1;j<=m;j++) b[ve[i]][j]=0;
	}
	ll c[20];
	memset(c,0,sizeof(c));
	for(int i=1;i<=m;i++){
		for(int j=1;j<=n;j++) c[i]+=b[j][i];
	}
	sort(c+1,c+m+1);
	for(int i=m;i>m-(k-ve.size());i--){
		ans+=c[i];
	}
	return ans;
}
int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            cin>>a[i][j];
            sum+=a[i][j];
        }
    }
    if(k>=n||k>=m){
        cout<<sum<<endl;
        return 0;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            r[i]+=a[i][j];
        }
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j<=n;j++){
            l[i]+=a[j][i];
        }
    }
    ll ans=0;
    for(int i=0;i<=(1<<n)-1;i++){
    	ans=max(ans,solve(i));
	}
	cout<<ans<<endl;
    return 0;
}