矩阵消除游戏
题目地址:
基本思路:
我们看到题目的范围非常小,因此我们考虑二进制枚举,
我们枚举每一行,然后我们在选择行确定的情况下去贪心的选择列就好了,
代码量可能比较多,但是没什么思维难度,具体可以看代码注释。
参考代码:
#pragma GCC optimize(2) #pragma GCC optimize(3) #include <bits/stdc++.h> using namespace std; #define IO std::ios::sync_with_stdio(false); cin.tie(0) #define int long long #define ll long long #define SZ(x) ((int)(x).size()) #define all(x) (x).begin(), (x).end() #define rep(i, l, r) for (int i = l; i <= r; i++) #define per(i, l, r) for (int i = l; i >= r; i--) #define mset(s, _) memset(s, _, sizeof(s)) #define pb push_back #define pii pair <int, int> #define mp(a, b) make_pair(a, b) #define INF 0x3f3f3f3f inline int read() { int x = 0, neg = 1; char op = getchar(); while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); } while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); } return neg * x; } inline void print(int x) { if (x < 0) { putchar('-'); x = -x; } if (x >= 10) print(x / 10); putchar(x % 10 + '0'); } const int maxn = 20; int n,m,k; int a[maxn][maxn]; int h_sum[maxn],l_sum[maxn],del[maxn]; signed main() { IO; cin >> n >> m >> k; rep(i,0,n-1) rep(j,0,m-1) cin >> a[i][j]; for(int i = 0 ; i < n ; i++){ // 行前缀和; h_sum[i] = 0; for(int j = 0 ; j < m ; j++) h_sum[i] += a[i][j]; } for(int j = 0 ; j < m ; j++) { // 列前缀和; l_sum[j] = 0; for (int i = 0; i < n; i++) l_sum[j] += a[i][j]; } int ans = 0; for(int p = 0 ; p < (1 << n) ; p++){ // 枚举行的选择状态; int cnt = __builtin_popcount(p); // 选择了的行数; if(cnt > k) continue; // 如果大于k,那么不可能; int res = 0; mset(del,0); // del记录算行时,每列被计算的部分; for(int i = 0 ; i < n ; i++) { if ((p >> i) & 1) { res += h_sum[i]; // 行的权值贡献; for (int j = 0; j < m; j++) del[j] += a[i][j]; } } vector<int> now; for(int j = 0 ; j < m ; j++) { now.push_back(l_sum[j] - del[j]); // 现在每列剩余的权值,为整列权值减行中被计算部分; } sort(all(now),greater<>()); // 从大到小排序,贪心取最多的; int z = k - cnt; for(int i = 0 ; i < z && i < SZ(now) ; i++) res += now[i]; ans = max(ans,res); } cout << ans << '\n'; return 0; }