参考链接
- https://leetcode-cn.com/problems/score-after-flipping-matrix/
题目描述
有一个二维矩阵 A 其中每个元素的值为 0 或 1 。
移动是指选择任一行或列,并转换该行或列中的每一个值:将所有 0 都更改为 1,将所有 1 都更改为 0。
在做出任意次数的移动后,将该矩阵的每一行都按照二进制数来解释,矩阵的得分就是这些数字的总和。
返回尽可能高的分数。
解题思路
贪心策略,要数字最大,最高位要为1,所以先通过移动哪些最高位不是1的行,使所有数的最高位都为1;接下来,就只能移动列了,只需要看每一列里面,0和1的数量,如果0比较多,就移动该列。
代码
class Solution {
public:
int matrixScore(vector<vector<int>>& A) {
for (int row = 0; row < A.size(); row ++)
{
if (A[row][0] == 0)
{
for (int col = 0; col < A[0].size(); col ++)
{
A[row][col] ^= 1;
}
}
}
int res = A.size() * (1 << (A[0].size() - 1));
for (int col = 1; col < A[0].size(); col ++)
{
int num_zero = 0;
for (int row = 0; row < A.size(); row ++)
{
if (A[row][col] == 0)
{
num_zero ++;
}
}
if (num_zero <= A.size() / 2)
{
num_zero = A.size() - num_zero;
}
res += num_zero * (1 << (A[0].size() - col - 1));
}
return res;
}
};