LeetCode算法題目

題目

給定一個二進制矩陣A，我們想先水平翻轉(zhuǎn)圖像，然后反轉(zhuǎn)圖像并返回結(jié)果。

水平翻轉(zhuǎn)圖片就是將圖片的每一行都進行翻轉(zhuǎn)，即逆序。例如，水平翻轉(zhuǎn) [1, 1, 0] 的結(jié)果是 [0, 1, 1]。

反轉(zhuǎn)圖片的意思是圖片中的 0 全部被 1 替換， 1 全部被 0 替換。例如，反轉(zhuǎn) [0, 1, 1] 的結(jié)果是 [1, 0, 0]。
示例 1:

輸入: [[1,1,0],[1,0,1],[0,0,0]]
輸出: [[1,0,0],[0,1,0],[1,1,1]]
解釋: 首先翻轉(zhuǎn)每一行: [[0,1,1],[1,0,1],[0,0,0]]；
     然后反轉(zhuǎn)圖片: [[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]]
解釋: 首先翻轉(zhuǎn)每一行: [[0,0,1,1],[1,0,0,1],[1,1,1,0],[0,1,0,1]]；
     然后反轉(zhuǎn)圖片: [[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

解答

Python3

方法一：

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

方法二：

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 x : int(x!=1), line)))
        return result