有一个二维矩阵 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
贪心。
每一行的数字要尽可能大,因此,遍历每一行,若行首元素为 0,则将该行每个元素进行翻转,即 grid[i][j] ^= 1
;
接着,遍历每一列,统计列中元素为 1 的个数 cnt
,若 cnt
(1 的个数) 比 m - cnt
(0 的个数) 小,则将该列进行翻转。实际过程中,并不需要对列进行翻转,只需要取 max(cnt, m - cnt)
,即表示 1 的个数,再乘上该位的大小 n - j - 1
,即求得当前列的大小。累加每一列大小即可。
class Solution:
def matrixScore(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid[0])
for i in range(m):
if grid[i][0] == 0:
for j in range(n):
grid[i][j] ^= 1
res = 0
for j in range(n):
cnt = 0
for i in range(m):
cnt += grid[i][j]
res += max(cnt, m - cnt) * (1 << (n - j - 1))
return res
class Solution {
public int matrixScore(int[][] grid) {
int m = grid.length, n = grid[0].length;
for (int i = 0; i < m; ++i) {
if (grid[i][0] == 0) {
for (int j = 0; j < n; ++j) {
grid[i][j] ^= 1;
}
}
}
int res = 0;
for (int j = 0; j < n; ++j) {
int cnt = 0;
for (int i = 0; i < m; ++i) {
cnt += grid[i][j];
}
res += Math.max(cnt, m - cnt) * (1 << (n - j - 1));
}
return res;
}
}
class Solution {
public:
int matrixScore(vector<vector<int>>& grid) {
int m = grid.size(), n = grid[0].size();
for (int i = 0; i < m; ++i)
{
if (grid[i][0] == 0)
{
for (int j = 0; j < n; ++j) grid[i][j] ^= 1;
}
}
int res = 0;
for (int j = 0; j < n; ++j)
{
int cnt = 0;
for (int i = 0; i < m; ++i) cnt += grid[i][j];
res += max(cnt, m - cnt) * (1 << (n - j - 1));
}
return res;
}
};
func matrixScore(grid [][]int) int {
m, n := len(grid), len(grid[0])
for i := 0; i < m; i++ {
if grid[i][0] == 0 {
for j := 0; j < n; j++ {
grid[i][j] ^= 1
}
}
}
res := 0
for j := 0; j < n; j++ {
cnt := 0
for i := 0; i < m; i++ {
cnt += grid[i][j]
}
res += max(cnt, m-cnt) * (1 << (n - j - 1))
}
return res
}
func max(a, b int) int {
if a > b {
return a
}
return b
}