2024. 6. 18. 21:16ㆍ개발의 흔적/코딩테스트
제목 : 카드 역배치
1부터 20까지 숫자가 하나씩 쓰인 20장의 카드가 아래 그림과 같이 오름차순으로 한 줄로 놓 여있다. 각 카드의 위치는 카드 위에 적힌 숫자와 같이 1부터 20까지로 나타낸다.

이제 여러분은 다음과 같은 규칙으로 카드의 위치를 바꾼다: 구간 [a, b] (단, 1 ≤ a ≤ b ≤ 20)가 주어지면 위치 a부터 위치 b까지의 카드를 현재의 역순으로 놓는다.
예를 들어, 현재 카드가 놓인 순서가 위의 그림과 같고 구간이 [5, 10]으로 주어진다면, 위치 5부터 위치 10까지의 카드 5, 6, 7, 8, 9, 10을 역순으로 하여 10, 9, 8, 7, 6, 5로 놓는다. 이제 전체 카드가 놓인 순서는 아래 그림과 같다.

이 상태에서 구간 [9, 13]이 다시 주어진다면, 위치 9부터 위치 13까지의 카드 6, 5, 11, 12, 13을 역순으로 하여 13, 12, 11, 5, 6으로 놓는다. 이제 전체 카드가 놓인 순서는 아래 그림 과 같다.

오름차순으로 한 줄로 놓여있는 20장의 카드에 대해 10개의 구간이 주어지면, 주어진 구간의 순서대로 위의 규칙에 따라 순서를 뒤집는 작업을 연속해서 처리한 뒤 마지막 카드들의 배치 를 구하는 프로그램을 작성하시오.
<입력설명>
총 10개의 줄에 걸쳐 한 줄에 하나씩 10개의 구간이 주어진다. i번째 줄에는 i번째 구간의 시 작 위치 ai와 끝 위치 bi가 차례대로 주어진다. 이때 두 값의 범위는 1 ≤ ai ≤ bi ≤ 20이다.
<출력설명>
1부터 20까지 오름차순으로 놓인 카드들에 대해, 입력으로 주어진 10개의 구간 순서대로 뒤집 는 작업을 했을 때 마지막 카드들의 배치를 한 줄에 출력한다.
5 10
9 13
1 2
3 4
5 6
1 2
3 4
5 6
1 20
1 20
출력예제
1 2 3 4 10 9 8 7 13 12 11 5 6 14 15 16 17 18 19 20
일주일 전에 작성한 내 코드
import sys
input = sys.stdin.readline
num_list = [i for i in range(21)]
index_list = [list(map(int, input().split())) for _ in range(10)]
for index in index_list:
start, end = index[0], index[1]
new_part = num_list[start: end+1]
for i in range(start, end+1):
num_list[i] = new_part.pop()
for i in num_list:
if i != 0:
print(i, end=" ")
numlist 의 range를 21로 설정한 이유는 아래 for문에서 start와 end index를 매번 계산해 주기 귀찮아서이다.
아무튼 slicing한 구간을 reverse시키고 다시 그부분을 list에 update하는 문법이 익숙지 않아 길어진 코드.
import sys
input = sys.stdin.readline
num_list = [i for i in range(21)]
index_list = [list(map(int, input().split())) for _ in range(10)]
for start, end in index_list:
num_list[start: end+1] = num_list[start: end+1][::-1]
# for index in index_list:
# start, end = index[0], index[1]
# new_part = num_list[start: end+1]
# for i in range(start, end+1):
# num_list[i] = new_part.pop()
for i in num_list:
if i != 0:
print(i, end=" ")
num_list[start: end+1] = num_list[start: end+1][::-1]
#num_list의 [start~end]를, num_list[start~end]의 reverse로 update 하겄다
이러한 detail한 python list slicing 문법들에 익숙해지도록 하자
'개발의 흔적 > 코딩테스트' 카테고리의 다른 글
| [구현] (3-4) list 합치기: merge sort (0) | 2024.06.18 |
|---|---|
| [구현] (3-2) 숫자만 추출 (0) | 2024.06.18 |
| [구현] (3-1) 회문 문자열 검사: for else문 / reverse ([::-1]) (1) | 2024.06.17 |
| [구현] (3-6) 격자판의 합 : 한 iteration 최대한 써먹기 (0) | 2024.06.17 |
| [two pointer] (3-5) list의 구간을 두개의 포인터로, 시간복잡도 줄이기 (0) | 2024.06.17 |