Python으로 푸는 LeetCode 832. Flipping an Image (Easy)

주어진 2차원 배열의 원소를 각각 뒤집고, 1을 0으로 0을 1로 변환시키는 코드를 짜시오.

LeetCode에서 푼 문제 리스트 보기

LeetCode에서 문제 보기

github에서 코드 보기

문제 풀이

Python의 reverse() 메소드를 사용하여 주어진 리스트 자체를 뒤집을 수 있다. (reversed()를 사용하면 새로운 리스트로 반환됨) 그 다음 리스트의 원소인 작은 리스트를 순회하면서 1을 0으로 0을 1로 바꿔주면 된다.

파이썬 코드

class Solution:
    def flipAndInvertImage(self, A):
        """
        Rumtime : faster than 74.47% of Python3
        Memory Usage : less than 5.63% of Python3
        """
        for item in A:
            item.reverse()
 
            for i in range(len(item)):
                if item[i]:
                    item[i] = 0
                else:
                    item[i] = 1
        return A