要求
给定一个二进制矩阵 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]]
核心代码
class Solution:
def flipAndInvertImage(self, image: List[List[int]]) -> List[List[int]]:
for i,x in enumerate(image):
image[i] = x[::-1]
for row in image:
for i,x in enumerate(row):
row[i] = 1 - x
return image
另一解法
class Solution:
def flipAndInvertImage(self, image: List[List[int]]) -> List[List[int]]:
for index,item in enumerate(image):
image[index].reverse()
image[index] = [(1-j) for j in image[index]]
return image
解题思路:使用两趟扫描的方式,第一趟我们先将行进行一个翻转,第二趟,我们针对每一个元素进行取反,得到翻转后的图像,就是规则这么说我们怎么做。第二种解法:我们使用一趟扫描法,在一趟循环中,即完成了行的翻转,又完成了行中元素的取反操作。