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. 一定要保证第一列全为1,因为1000 比 0111大,因此第一列一定要为1

2. 完成第一步之后,开始对列进行调整,而不是对行进行调整,因为对行调整要不然等同一次变换要不然与原始一致,如果对该行调整偶数次,则恢复原样,若调整奇数次,则等同于调整1次。

3. 对列调整就可以根据0和1的个数来判断是否进行调整了,因为同一列的数都是同等级的,即在二进制中占据相同的位级。

4. 最后对行求和即可。
class Solution {
public:
    int matrixScore(vector>& A) {
        int m = A.size();       // m行n列
        if (m == 0) return 0;
        int n = A[0].size();
        for (int i = 0; i < m; ++i){  // 调整第一列全为1     
            if (A[i][0] == 0){
                for (int j = 0; j < n; ++j){
                    A[i][j] = 1 - A[i][j];
                }
            }
        }

        for (int i = 1; i < n; ++i){   // 从第二列开始调整
            int cnt = 0;
            for (int j = 0; j < m; ++j){
                cnt += A[j][i];
            }
            if (cnt <= m / 2){
                for (int j = 0; j < m; ++j){
                    A[j][i] = 1 - A[j][i];
                }               
            }
        }
        int res = 0;
        for (int i = 0; i < m; ++i){   // 求和
            int cnt = 0;
            for (int j = 0; j < n; ++j){
                cnt = (cnt << 1) + A[i][j];    
            }
            res += cnt;
        }    
        return res;
    }
};

 

你可能感兴趣的:(Leetcode)