Leetcode 861.翻转矩阵后的得分
1 题目描述(Leetcode题目链接)
有一个二维矩阵 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
2 题解
贪心策略:最高位都变成1,其余列1的个数为该列中0和1出现次数的最大值。
class Solution:
def matrixScore(self, A: List[List[int]]) -> int:
m, n = len(A), len(A[0])
res = m*2**(n-1) # 第一列都是1
for i in range(m): # 反转每一行,使得第一列都是1
if A[i][0] == 0:
for j in range(n):
A[i][j] = 1 - A[i][j]
for j in range(1, n):
num = {0:0, 1:0}
for i in range(m):
num[A[i][j]] += 1
res += max(num[0], num[1])*2**(n - 1 - j)
return res