[백준] 10811번 바구니 뒤집기(python)

문제소개

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 순서대로 적혀져 있다. 바구니는 일렬로 놓여져 있고, 가장 왼쪽 바구니를 1번째 바구니, 그 다음 바구니를 2번째 바구니, ..., 가장 오른쪽 바구니를 N번째 바구니라고 부른다.

도현이는 앞으로 M번 바구니의 순서를 역순으로 만들려고 한다. 도현이는 한 번 순서를 역순으로 바꿀 때, 순서를 역순으로 만들 범위를 정하고, 그 범위에 들어있는 바구니의 순서를 역순으로 만든다.

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에는 바구니의 순서를 역순으로 만드는 방법이 주어진다. 방법은 i j로 나타내고, 왼쪽으로부터 i번째 바구니부터 j번째 바구니의 순서를 역순으로 만든다는 뜻이다. (1 ≤ i ≤ j ≤ N)

도현이는 입력으로 주어진 순서대로 바구니의 순서를 바꾼다.

출력

모든 순서를 바꾼 다음에, 가장 왼쪽에 있는 바구니부터 바구니에 적혀있는 순서를 공백으로 구분해 출력한다.

• 문제를 보면 도현이가 N개의 바구니를 가지고 있다. 이 바구니는 1부터 N번까지의 순서대로 번호가 부여되어 있다고 한다.

• 이 바구니를 M번 i번 바구니부터 j번 바구니의 순서를 역순으로 한다.

풀이 과정

• 입력의 첫 줄은 N,M값을 받아와 N값을 가지는 바구니를 리스트로 만들어 준다.

n, m = map(int, input().split())

baguni = [x for x in range(1,n+1)]

• 이 후 M번 순서가 바뀌므로 M값을 range로 가지는 for문을 생성하고 i,j의 값을 입력 받는다.

for i in range(m):
    fir, last = map(int, input().split())

• i번 바구니부터 j번 바구니까지의 순서를 역순으로 하기 위해서 reversed()를 이용하여 바구니 리스트의 i-1인덱스부터 j-1인덱스 까지의 순서를 역순으로 정렬한 ans라는 임시 리스트를 생성한다. 이 후 이를 바구니 리스트에 업데이트한다.

for i in range(m):
    fir, last = map(int, input().split())
    ans = [y for y in reversed(baguni[fir-1:last])]

    baguni[fir-1:last] = ans

• 출력형태가 리스트가 아니기에 join을 활용해서 출력합니다.

print(' '.join(str(z) for z in baguni))

아래와 같이 출력해도 무방합니다.

print(*baguni)

전체코드

n, m = map(int, input().split())

baguni = [x for x in range(1,n+1)]

for i in range(m):
    fir, last = map(int, input().split())
    ans = [y for y in reversed(baguni[fir-1:last])]

    baguni[fir-1:last] = ans

print(' '.join(str(z) for z in baguni))

파이썬의 내부 함수를 잘 알고 있어야 구현이 편하다는 것을 알게되었다.
또한 배열을 다루는 연습이 되었던 문제였다.
reversed를 사용하지 않고 푸는 방법도 고민을 해보면 좋을 것 같다.