LeetCode 861. 翻转矩阵后的得分

Posted 2021-01-09 hlk09

有一个二维矩阵 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

简单的贪心策略。先保证每一行开头都为0，然后保证每列尽可能多一些1。

class Solution {
public:
    int matrixScore(vector<vector<int>>& A) {
        if(A.size() == 0 || A[0].size() == 0) {
            return 0;
        }
        int row = A.size(), column = A[0].size(), sum = 0;
        for(int i = 0; i < row; i++) {
            if(A[i][0] == 0) {
                for(int j = 0; j < column; j++) {
                    A[i][j] = 1 - A[i][j];
                }
            }
        }
        for(int j = 0; j < column; j++) {
            int zero = 0, one = 0;
            for(int i = 0; i < row; i++) {
                if(A[i][j] == 0) {
                    zero++;
                } else {
                    one++;
                }
            }
            if(zero > one) {
                for(int i = 0; i < row; i++) {
                    A[i][j] = 1 - A[i][j];
                }
            }
        }
        for(int i = 0; i < row; i++) {
            for(int j = 0; j < column; j++) {
                sum += A[i][j] * pow(2, column - j - 1);
            }
        }
        return sum;
    }
};