LeetCode刷题日记-数组-面试题 832. 翻转图像

题目描述

题目链接：https://leetcode-cn.com/problems/flipping-an-image/

解题思路

列表反转可以用[::-1]，0/1的相互转换可以用与1进行异或。
例如 0 ^ 1= 1, 1 ^ 1=0
刚开始想到了两次for循环遍历，代码如下：

    for x in A:
        x = x[::-1]   #  先翻转
        for y in x:   #  然后与1进行异或，0/1转换
            y ^= 1
    return A

结果忘记了，如果对列表进行遍历，直接对其中的元素赋值，待遍历结束之后，原来列表中的元素的值并不会不发生改变。例如：

>> A = [1, 2, 3,4]
>> for x in A:  
	x = 1
>>A
>[1, 2, 3, 4]
试图将每个元素修改为1，想要的结果应该是[1, 1, 1, 1]。
但实际上这样子做并没有改变。
因为每次遍历时的x其实是一个临时变量，
值相同，但是地址不同。

因此，我们需要返回一个新的列表，倒不如直接用列表推导式得了！（来自评论区的灵感）
解题代码如下：

class Solution:
    def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
        # 注意生成是双重列表
        # return 接一个`外`列表
        # 一个y就是一个`内列表
        return [ [x^1 for x in y[::-1]] for y in A]