知识点:
1.数组整行翻转:list[::-1]
2.要求让 0 变成 1,1 变成 0有两种方法:1-x,异或运算x^1
方法一
先整行翻转,再逐项替换
class Solution:
def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
n = len(A)
for i in range(n):
A[i] = A[i][::-1]
for j in range(n):
A[i][j] = 1-A[i][j]
return A
方法二
一边翻转一边修改
class Solution:
def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
n = len(A)
for i in range(n):
for j in range((n+1)//2):
A[i][j], A[i][n-j-1] = 1-A[i][n-j-1], 1-A[i][j]
return A