luoheng
10/6/2019 - 8:06 AM

flipAndInvertImage

func flipAndInvertImage(A [][]int) [][]int {
    w, h := len(A), len(A[0])
    for i := 0; i < w; i++ {
        for j := 0; j <= h / 2; j++ {
            x, y := A[i][j], A[i][h-j-1]
            if x == y && j <= h - j - 1 {
                A[i][j] = 1 - x
                if j != h - j - 1 {
                    A[i][h-j-1] = 1 - y
                }
            }
        }
    }
    return A
}