目录
一、题目描述
有一个二维矩阵 A 其中每个元素的值为 0 或 1 。
移动是指选择任一行或列,并转换该行或列中的每一个值:将所有 0 都更改为 1,将所有 1 都更改为 0。
在做出任意次数的移动后,将该矩阵的每一行都按照二进制数来解释,矩阵的得分就是这些数字的总和。
返回尽可能高的分数。
示例:
输入:[[0,0,1,1],[1,0,1,0],[1,1,0,0]]
输出:39
解释:
转换为 [[1,1,1,1],[1,0,0,1],[1,1,1,1]]
0b1111 + 0b1001 + 0b1111 = 15 + 9 + 15 = 39
提示:
- 1 <= A.length <= 20
- 1 <= A[0].length <= 20
- A[i][j] 是 0 或 1
二、解题思路
这题除了贪心也想不到其他可行的方案了…
贪心的思路是:对每一行来说,行首必须为1才最大,所以行首不为1则需要翻转这一行;对每一列来说,每一列1的数目要尽可能的多,1的数目少于半数的列需要翻转。
三、代码实现
#include <bits/stdc++.h>
using namespace std;
//这是不改变矩阵值的方式
int matrixScore(vector<vector<int>>& A) {
int m = A.size();
int n = A[0].size();
//首列必须为1才能使得数字最大
//计算结果时按照列计算,把每一列的值相加即可
int res = m * (1 << n - 1);
//统计每一列中1的数目,然后相加
for (int j = 1; j < n; j++) {
int cntOfOne = 0;
for (int i = 0; i < m; i++) {
if (A[i][0] == 1) {
cntOfOne += A[i][j];
} else {
cntOfOne += (1 - A[i][j]);
}
}
cntOfOne = max(cntOfOne, m - cntOfOne);
res += cntOfOne * (1 << n - 1 - j);
}
return res;
}
//如果要改变矩阵值的话就是三步:
//1.先把行首不为0的行翻转;
//2.把每一列中1的数量少于一半的列翻转
//3.求和
//时间复杂度同样为O(mn)
int main() {
vector<vector<int>> A = { {0,0,1,1},{1,0,1,0,},{1,1,0,0} };
int value = matrixScore(A);
cout << value << endl;
return 0;
}