【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

思路

一开始学机器学习学傻了，竟然想用启发式来做，遍历每行每列，如果对行进行"移动"后该行值变大，或者对列进行"移动"后整体变大，那么就保留该操作，直到整体收敛 哭o(╥﹏╥)o

后来想到是贪心

对行进行“操作”的前提是使得行的值变大，而对于二进制计算值而言，首位为1时最大，如果再逆转，那么首位不为1的话，一定会变小
所以对行操作为保证首位为1

随后对列进行“操作”的前提是使得整体的值变大，而对列进行逆转，只会改变某行第i位的值，而每行第i位对整体的贡献的一样的
所以只需要保证第i位的1变多，那么整体就会变大

实现上的细节

列表表达式和条件表达式非常实用
grid[i] = [0 if (value == 1) else 1 for value in temp]
grid[i][j] = 0 if (grid[i][j] == 1) else 1
将二进制字符串转为十进制数
int(binary_str,2)

python实现

from typing import List

class Solution:
    def matrixScore(self, grid: List[List[int]]) -> int:
        """
        贪心
        先按照行遍历，将1尽可能放到左面，保证最左面的一定为1
        再按照列遍历，保证列上尽可能多的1
        """
        max_value = 0
        # 对行操作
        for i in range(len(grid)):
            if (grid[i][0] == 0):
                temp = grid[i]
                grid[i] = [0 if (value == 1) else 1 for value in temp]
        # 对列操作
        for j in range(len(grid[0])):
            counter_1 = 0
            counter_0 = 0
            for i in range(len(grid)):
                if (grid[i][j] == 1):
                    counter_1 += 1
                else:
                    counter_0 += 1
            if (counter_1 < counter_0):
                for i in range(len(grid)):
                    grid[i][j] = 0 if (grid[i][j] == 1) else 1
        # 求和
        for i in range(len(grid)):
            temp = [str(v) for v in grid[i]]
            binary_str = "".join(temp)
            binary_int = int(binary_str,2)
            max_value+=binary_int
        return max_value



# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    input_value=[[0,0,1,1],[1,0,1,0],[1,1,0,0]]
    s=Solution()
    print(s.matrixScore(input_value))