NowCoder 练习 —— 矩阵消除游戏（思维+枚举）

题目链接: 矩阵消除游戏

一、题目描述

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

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

二、输入描述:

• 第一行三个整数n,m,k
• 接下来n行每行m个整数表示矩阵中各个单元格的权值。
• 1≤n,m≤15
• 1≤a[i][j]≤1e6
• 1≤k≤n∗m

三、输出描述:

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

示例1：

• 输入
  3 3 2
  101 1 102
  1 202 1
  100 8 100
• 输出
  414

四、思路

• 这道题可以用搜索做也可以用枚举，我用的是枚举；
• 怎么枚举呢，对于该矩阵，用一个结构体数组存放该行下标和改行所有值的和，对于k > n或者m，直接能够对所有值取得。当 k < min(n, m)时，则需要开始枚举；
• 首先，对于行结构体数组，对他们从大到小进行排序，从0~k进行枚举，每次选择一行时，对改行进行标记，对于已选择i行，那么对k-i列进行选择，同样的方法选择最大的k-i列，更新ans的值；

五、Code

#include <bits/stdc++.h>

using namespace std;
#pragma GCC optimize(2)
#define ios std::ios::sync_with_stdio(false);cin.tie(0), cout.tie(0);
#define ll long long
const int Max = 1e6+3;
const int mod = 1e9+7;

int n, m, k;
ll a[20][20], cnt=0;
pair<int, int> row[20];
bool book[20];

bool cmp(pair<int, int> x, pair<int, int> y)	//自定义排序 
{
	return x.second > y.second;
}
bool cmp2(ll x, ll y)
{
	return x > y;
}

int main()
{
	ios
	
	cin >> n >> m >> k;
	for(int i=1; i<=n; i++)
	{
		row[i].first = i;
		row[i].second = 0;
		for(int j=1; j<=m; j++)
		{
			cin >> a[i][j];
			row[i].second += a[i][j];
		}
	}
	int sum = min(k, min(n, m));
	sort(row+1, row+n+1, cmp);
	for(int i=0; i<=n; i++)
	{
		ll col[20]={0}, ans=0;
		int tmp = sum-i;
		if(tmp > m||tmp < 0) continue;
		for(int j=1; j<=i; j++)
		{
			ans += row[j].second;
			book[row[j].first] = true;
		}
		for(int j=1; j<=m; j++)
		{
			for(int v=1; v<=n; v++)
			{
				if(book[v]) continue;
				col[j] += a[v][j];
			}
		}
		sort(col+1, col+m+1, cmp2);
		for(int j=1; j<=tmp; j++)
			ans += col[j];
		cnt = max(cnt, ans); 
	}
	cout << cnt << endl;
	return 0;
}

---

蒟蒻一只，欢迎指正