有一个二维矩阵 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
的个数多于0
的个数即可。
class Solution {
public:
int matrixScore(vector<vector<int>>& A) {
int m = A.size(), n = A[0].size();
// 先通过行翻转将第一列全翻转为1
for(int i = 0; i < m; i++){
if(A[i][0] == 0){
for(int j = 0; j < n; j++){
A[i][j] ^= 1;
}
}
}
// 通过列翻转使除第一列以外的所有列中 1 的个数多于 0 的个数
for(int j = 1; j < n; j++){
// 统计列中 1 的个数
int cnt = 0;
for(int i = 0; i < m; i++){
cnt += A[i][j];
}
if(cnt <= (m >> 1)){
for(int i = 0; i < m; i++){
A[i][j] ^= 1;
}
}
}
// 求和
int sum = 0;
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
if(A[i][j] == 1)
sum += (1 << (n - j - 1));
}
}
return sum;
}
};