[Python] 백준 1201 & 백준 1158

#백준 1201

from collections import deque

n, m = map(int, input().split(" "))
targets = list(map(int, input().split(' ')))

dq = deque(range(1, n+1))

# 10 3 [1,2,3,4,5,6...]
# 2 9 5
cnt = 0
for i in targets:
    if dq[0] == i:
        dq.popleft()
        continue
    index = dq.index(i) #찾고자 하는 값의 인덱스 위치가 9이면
    if index <= len(dq)//2: #3 < 5
        dq.rotate(-index)
        cnt += index
    else:
        dq.rotate(len(dq)-index)
        cnt += len(dq)-index

    dq.popleft()
print(cnt)

 

#백준 1158 요세푸스 문제

from collections import deque
n, m = map(int, input().split(' '))
numbers= [i for i in range(1, n+1)]
dq = deque(numbers)
answer = []

while dq:
    dq.rotate(-(m-1))
    num = dq.popleft()
    answer.append(num)

print(f'<{", ".join(str(x) for x in answer)}>')

 

 📝 Points

• deque.append(item): item을 데크의 오른쪽 끝에 삽입한다.
• deque.appendleft(item): item을 데크의 왼쪽 끝에 삽입한다.
• deque.pop(): 데크의 오른쪽 끝 엘리먼트를 가져오는 동시에 데크에서 삭제한다.
• deque.popleft(): 데크의 왼쪽 끝 엘리먼트를 가져오는 동시에 데크에서 삭제한다.
• deque.remove(item): item을 데크에서 찾아 삭제한다.
• deque.index(x[, start[, stop]]): 큐에서 x의 위치를 반환한다.
• deque.rotate(k): 큐 안의 원소를 우측(반시계방향)으로 k만큼 이동, k가 음수일 경우 왼쪽(시계방향)으로 k만큼 이동
• print(f'<',', '.join(str(x) for x in answer),'>',sep="") : <> 과 출력값을 sep=""을 통해서 빈칸없이 붙이고 ', '.join(str(x) for x in answer) 를 통해 answer있는 원소(x)들을 문자(str)로 만들고 띄어쓰기와 콤마를 붙인다.