yankewei / LeetCode

LeetCode 问题的解决方法
MIT License
6 stars 0 forks source link

翻转图像 #116

Open yankewei opened 3 years ago

yankewei commented 3 years ago

给定一个二进制矩阵 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]]

提示:

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/flipping-an-image 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

yankewei commented 3 years ago

模拟 + 双指针优化

一个数组的首尾交换可以用双指针来进行交换,但是由于题目中说了,为0或者1,所以如果首尾两个指针的值不一样,翻转后再反转其实没有变化,所以如果两个指针的值不同,不进行任何操作,如果相同,直接取反即可,省去了交换的步骤。

func flipAndInvertImage(A [][]int) [][]int {
    for i := 0; i< len(A); i++ {
        left, right := 0, len(A[i])-1 
        for left < right {
            if A[i][left] == A[i][right] {
                A[i][left] ^= 1
                A[i][right] ^= 1
            }
            left++
            right--
        }
        // 可能left和right相同
        if left == right {
            A[i][left] ^= 1
        }
    return A
}