LeetCode每日一题(Python实现)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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/score-after-flipping-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:

遍历每一行,若A[i][0] == 0 则反转改行。
遍历每一列,若该列=0的元素个数 > 该列 = 1的元素个数 则反转该列。
每一行二进制相加

代码:

from typing import *
class Solution:
          def matrixScore(self,A: List[List[int]]) -> int:
                # n行 m列
                n = len(A)
                m = len(A[0])
                for i in range(n):
                    if A[i][0] == 0:
                        for j in range(m):
                            if A[i][j] == 1:
                                A[i][j] = 0
                            else:
                                A[i][j] = 1
                # i 列 j 行
                for i in range(m):
                    count0 = 0
                    count1 = 0
                    for j in range(n):
                        if A[j][i] == 0:
                            count0 += 1
                        else :
                            count1 += 1
                    if count0 > count1:
                        for j in range(n):
                             if A[j][i] == 1:
                                A[j][i] = 0
                             else:
                                A[j][i] = 1
                print(A)
                num = 0
                for i in range(n):
                    temp = 0
                    for j in range(m):
                        temp = temp*2 + A[i][j]
                    num += temp
                return num
ans = Solution()
n = int(input())
m = int(input())
A = [[0,0,1,1],[1,0,1,0],[1,1,0,0]]
print(ans.matrixScore(A))

猜你喜欢

转载自blog.csdn.net/weixin_44795952/article/details/110847746