有一个二维矩阵 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,只有保证最高位为1,才能保证这一行得分最大。(贪心思想,见到首位不为1就翻转)
然后从第二列开始扫描,使每一列拥有1最多(如果1的个数小于0就对该列进行该列翻转)
class Solution {
public:
int matrixScore(vector<vector<int>>& A) {
if(A.empty()) return 0;
int n = A.size();
int m = A[0].size();
for(int i = 0; i < n; i++){
if(A[i][0] == 0){
for(int j = 0; j < m; j++){
A[i][j] ^= 1;
}
}
}
int sum = n * pow(2, m-1);
for(int i = 1; i < m; i++){
int cnt = 0;
for(int j = 0; j < n; j++){
if(A[j][i] == 1) cnt++;
}
// 1的个数小于0,这里只需要统计1的个数,直接统计即可,不用真的翻转
if(cnt <= n / 2) cnt = n - cnt; // 等于号要取
sum += cnt * pow(2, m-i-1);
}
return sum;
}
};