time O(N) space O(1)
class Solution:
def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
row = len(A)
col = len(A[0])
# reverse
for i in range(row):
for j in range(col//2):
A[i][j] , A[i][col-1-j] = A[i][col-1-j] , A[i][j]
# invert
for i in range(row):
for j in range(col):
if A[i][j] == 1: # or A[i][j] = 1 - A[i][j]
A[i][j] = 0
else:
A[i][j] = 1
return A