CodingTest/Baekjoon

[백준/Python] 10811번 : 바구니 뒤집기

JuniTech 2023. 5. 17. 09:00
728x90
시간 제한  메모리 제한 문제 티어 정답 비율
1초 256MB
브론즈 II
58.463%

📜 문제

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

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

바구니의 순서를 어떻게 바꿀지 주어졌을 때, M번 바구니의 순서를 역순으로 만든 다음, 바구니에 적혀있는 번호를 가장 왼쪽 바구니부터 출력하는 프로그램을 작성하시오.

 

📥입력

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

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

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

 

📤출력

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

 

💡풀이

이번 문제의 포인트는 '반복문과 리스트의 슬라이스(slice) 기능을 적절히 활용하였는가?' 이다.
 
먼저 정수 1부터 N까지 들어있는 리스트를 만들어 basket_list 변수에 저장을 하고,
 
M만큼 반복을 하여 정수 i와 j를 입력받게 한다.
 
그러고 나서 임시 변수인 temp_basket_list를 만들어 인덱스 i-1 ~ j 에 있는 요소들을 뒤집어 그 부분만을 리스트로 저장한다.
 
그 다음에  basket_list[i-1:j] = temp_basket_list 를 통해서 i-1 ~ j 부분이 역으로 저장된 temp_pasket_list를 basket_list의 인덱스 i-1 ~ j 에 있는 요소들과 바꾼다.
 
나는 맨 처음에 어떻게 하면 리스트의 일부분을 바꿀 수 있는지 헷갈렸었다. 그래서 이것 저것 코딩해보다가 아래와 같이 그냥 리스트끼리는 해당 부분에 넣으면 알아서 리스트 껍데기가 벗겨지고 요소로 들어가지는 걸 깨달았다.
basket_list = [1, 2, 3, 4, 5]
temp_basket_list = list(reversed(basket_list[0:2]))    # temp_basket_list = [2, 1]
basket_list[i-1:j] = temp_basket_list     # basket_list = [2, 1, 3, 4, 5]
 

 

아래는 정답 코드이다.

import sys

N, M = map(int, sys.stdin.readline().split())

basket_list = [n for n in range(1, N+1)]

for k in range(M):
  i, j = map(int, sys.stdin.readline().split())

  temp_basket_list = list(reversed(basket_list[i-1:j]))
  basket_list[i-1:j] = temp_basket_list

print(*basket_list)