백준 알고리즘 11586번 지영 공주님의 마법 거울(python)

은근히 풀기 귀찮은 문자열 처리 문제다.

우선 공주의 모습을 n*n 형태의 리스트에 저장한다.

그 후 K가 2이면 li[i][::-1]을 통해 각 행들을 좌우반전시켜주고

K가 3이면 li[::-1]을 통해 상하반전시켜준다.

(출력해서 볼 때는 상하반전된 것처럼 보이지만 사실 리스트가 좌우반전된 상황이다)

마지막으로 각 행을 순서대로 출력해주면 된다.

n = int(input())
li = []
for i in range(n):
    li.append(input())
K = int(input())
if K == 2:
    for i in range(n):
        li[i] = li[i][::-1]
elif K == 3:
    li = li[::-1]
for i in range(n):
    print(li[i])