728x90
도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 순서대로 적혀져 있다. 바구니는 일렬로 놓여져 있고, 가장 왼쪽 바구니를 1번째 바구니, 그 다음 바구니를 2번째 바구니, ..., 가장 오른쪽 바구니를 N번째 바구니라고 부른다.
도현이는 앞으로 M번 바구니의 순서를 역순으로 만들려고 한다. 도현이는 한 번 순서를 역순으로 바꿀 때, 순서를 역순으로 만들 범위를 정하고, 그 범위에 들어있는 바구니의 순서를 역순으로 만든다.
바구니의 순서를 어떻게 바꿀지 주어졌을 때, M번 바구니의 순서를 역순으로 만든 다음, 바구니에 적혀있는 번호를 가장 왼쪽 바구니부터 출력하는 프로그램을 작성하시오.
내가 짠 코드는 아래와 같다.
새 리스트를 만들어서 뒤집은 후 기존 리스트 위치에 집어넣는 방법이다
지금까지 C 로 코딩테스트 풀이를 했었기 때문에 그 버릇이 안빠진것 같다
n, m = map(int, input().split())
# 개수 n 횟수 m
bucket = []
for i in range(0,n):
bucket.append(i+1)
for t in range(0,m):
i, j = map(int, input().split())
new = []
for p in range(i-1,j):
new.append(bucket[p])
new.reverse()
idx=0
for p in range(i-1,j):
bucket[p] = new[idx]
idx+=1
for i in range(0,n):
print(bucket[i], end=" ")
더 간결하게 코드를 짤 수 있는 방법을 찾아 Chat GPT에게 물어보니 아래와 같은 코드를 출력했다
n, m = map(int, input().split())
# 개수 n 횟수 m
bucket = list(range(1, n+1))
for _ in range(m):
i, j = map(int, input().split())
bucket[i-1:j] = reversed(bucket[i-1:j])
for num in bucket:
print(num, end=" ")
파이썬은 tmp를 쓰지 않고도 이런 식으로 교체가 가능하다는걸 알게 되었다
또한 리스트를 생성하는 한줄짜리 코드가 있다는 사실을 깨달았다
C 하던 버릇이 남아있어서 하나하나 리스트에 집어넣고 있었는데.... 왜 코테를 파이썬으로 보라고 하는지 알것같았다
bucket = list(range(1, n+1))
bucket[i-1:j] = reversed(bucket[i-1:j])