leetcode 861. 翻转矩阵后的得分

目录

• 题目描述：
• 示例：
• 解法：

题目描述：

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

---

解法：

class Solution {
public:
    int matrixScore(vector<vector<int>>& A) {
        int m = A.size();
        int n = A[0].size();
        int multi = 1;
        int res = 0;
        // 1. flip row
        for(int i = 0; i < m; i++){
            if(A[i][0] == 0){
                for(int j = 0; j < n; j++){
                    A[i][j] = 1 - A[i][j];
                }
            }
        }
        
        // 2. flip column
        for(int i = n-1; i >= 0; i--){
            int val = 0;
            for(int j = 0; j < m; j++){
                val += A[j][i];
            }
            // cout<<multi<<", "<<max(val, m-val)<<endl;
            res += multi*max(val, m - val);
            multi <<= 1;
        }
        return res;
    }
};