力扣刷题记录#数组#简单#832翻转图像

题目描述

给定一个二进制矩阵 A，先水平翻转图像，然后反转图像并返回结果。
水平翻转图片就是将图片的每一行都进行翻转，即逆序。例如，水平翻转 [1, 1, 0] 的结果是 [0, 1, 1]。
反转图片的意思是图片中的 0 全部被 1 替换， 1 全部被 0 替换。例如，反转 [0, 1, 1] 的结果是 [1, 0, 0]。

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

示例

输入: [[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]]

输入: [[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]]

解答

简单办法：遍历图像的每一行，对于每一行，用首尾指针来进行翻转，同时反转

class Solution(object):
    def flipAndInvertImage(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        row = len(A)
        col = len(A[0])
        
        # 遍历行
        for i in range(0,row):
            
            # 对每一行，用首尾双指针来交换元素，同时进行元素翻转
            l = 0
            r = col-1
            
            # 考虑到偶数的情况，要翻转每个元素，所以循环要取到等号
            while l<=r:
                A[i][l],A[i][r] = A[i][r],A[i][l]
                
                # 如果是偶数的情况，A[i][l]和A[i][r]都进行翻转，则最中间的数又变回原来的数
                # 所以要分情况翻转
                if l != r:
                    A[i][l] = 1 - A[i][l]
                    A[i][r] = 1 - A[i][r]
                if l==r:
                    A[i][l] = 1 - A[i][l]
                l += 1
                r -= 1
                
        return A