[구현] (3-3) list reverse의 마법 [:: -1]

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 문법들에 익숙해지도록 하자