「翻转图像」python之leetcode刷题|003

镇楼

题目

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

  • 1 <= A.length = A[0].length <= 20
  • 0 <= A[i][j] <= 1

解答

这道题是数组操作,遇上python用列表切片操作完美解决,同时复习上一题的位运算。

class Solution:
    def flipAndInvertImage(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        for i in range(len(A)):
            for j in range(len(A[0])):
                A[i][j] = A[i][j]^1
            A[i] = A[i][::-1]

        return A

用两层for循环来操作二维数组里的元素,进行异或位运算。再用python中列表的切片操作翻转列表元素。
看一下执行结果

执行结果

执行结果


看一下网友代码:

class Solution:
    def flipAndInvertImage(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        result = []
        for line in A:
            line = line[::-1]
            result.append(list(map(lambda a : int(a!=1),line)))
        return result

刚开始也是想用map()函数,把元素遍历出来,得到结果的数字,没有生成列表。网友代码是先把元素逆序,然后再进行翻转。

  • 这里用到了lambda匿名函数,这个函数的的作用是翻转数字。
  • 外层嵌套的map()函数接受两个参数,一个是匿名函数lambda,一个是列表。
  • 因为map()函数是惰性,所以用list函数求出列表。
  • 你有更好的解答方法吗,公众号sixkery,欢迎一起交流。

猜你喜欢

转载自blog.csdn.net/sixkery/article/details/81537287
今日推荐