832.翻转图像:对矩阵A的每行进行逆序翻转之后,把A的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]]
思路:为了省时间,对A的每一行,做一个类似指针的东西,最左边的left和最右边的right,从两头开始往中间走,交换两头的元素就可以实现逆转了,交换完了之后进行反转,0变1,1变0
代码:
class Solution:
def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
m = len(A)
n = len(A[0])
for i in range(m):
left = 0
right = n-1
#交换最左边和最右边的,往里缩进
while left < right:
temp = A[i][left]
A[i][left] = A[i][right]
A[i][right] = temp
left += 1
right -= 1
#反转
for j in range(n):
if A[i][j] == 0:
A[i][j] = 1
else:
A[i][j] = 0
return A