leetcode刷题——832. 翻转图像 python3

题目描述：
给定一个二进制矩阵 A，我们想先水平翻转图像，然后反转图像并返回结果。

水平翻转图片就是将图片的每一行都进行翻转，即逆序。例如，水平翻转 [1, 1, 0] 的结果是 [0, 1, 1]。

反转图片的意思是图片中的 0 全部被 1 替换， 1 全部被 0 替换。例如，反转 [0, 1, 1] 的结果是 [1, 0, 0]。
示例 1：

输入: [[1,1,0],[1,0,1],[0,0,0]]
输出: [[1,0,0],[0,1,0],[1,1,1]]
解释: 首先翻转每一行: [[0,1,1],[1,0,1],[0,0,0]]；
然后反转图片: [[1,0,0],[0,1,0],[1,1,1]]

示例 2：

输入: [[1,1,0,0],[1,0,0,1],[0,1,1,1],[1,0,1,0]]
输出: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]]
解释: 首先翻转每一行: [[0,0,1,1],[1,0,0,1],[1,1,1,0],[0,1,0,1]]；
然后反转图片: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]]

说明:

• 1 <= A.length = A[0].length <= 20

• 0 <= A[i][j] <= 1

思路：
先将题目所给数组中每一个列表中的数值都利用函数翻转。
这是第一步的翻转
再利用下标，将数组中每一个数值都“1”变“0”，“0”变“1”

代码

class Solution:
    def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
        b = []
        for i in A:
            i.reverse()
            b.append(i)
        for i in range(0,len(b)):
            for j in range(0,len(b[i])):
                if b[i][j] == 1:
                    b[i][j] = 0
                elif b[i][j] == 0:
                    b[i][j] = 1
        return b