[BOJ 1700🥇1] 멀티탭 스케줄링 (Python)

https://www.acmicpc.net/problem/1700

1700번: 멀티탭 스케줄링

문제

기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.

예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면,

1. 키보드
2. 헤어드라이기
3. 핸드폰 충전기
4. 디지털 카메라 충전기
5. 키보드
6. 헤어드라이기

키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다.

입력

첫 줄에는 멀티탭 구멍의 개수 N (1 ≤ N ≤ 100)과 전기 용품의 총 사용횟수 K (1 ≤ K ≤ 100)가 정수로 주어진다. 두 번째 줄에는 전기용품의 이름이 K 이하의 자연수로 사용 순서대로 주어진다. 각 줄의 모든 정수 사이는 공백문자로 구분되어 있다.

출력

하나씩 플러그를 빼는 최소의 횟수를 출력하시오.

예제 입력 1

2 7
2 3 2 3 1 2 7

예제 출력 1

2

 

⭕ CODE

from collections import deque

n,k = map(int,input().split())

arr = list(map(int,input().split()))

# 중간에 중복을 제거하기위한 new를 deque로 만들기
new = deque()
# arr의 첫번째 값 넣어주기
new.append(arr[0])

# [1,2,2,2,3,3,4,5,1,1]을
# [1,2,3,4,5,1]로 중간에 중복을 제거해주는 작업
for i in range(1,k):
	# 중복이 아니라면
    if new[-1] != arr[i]:
    	# new에 담아주기
        new.append(arr[i])

# 지금 멀티탭에 꼽혀있는 전기용품을 알려주는 tap
tap = deque()
# tap에 n개의 가전제품을 넣기
while True:
    tap.append(new.popleft())
    if len(tap) == n:
        break

# 바로 다음에 할 가전제품이
# 멀티탭에 꼽혀있으면 제거해준다
while new and new[0] in tap:
    new.popleft()

# 몇번 뽑았는지 확인하는 count
count = 0
# new가 다 비어야지 모든 전기용품 다 쓴거
while new:
	# 제일 나중에 쓸 가전제품을 찾아야한다.
    # 그리고 그거를 뽑아야지 가장 효율적
    # last_index[0] : 값
    # last_index[1] : 어떤 숫자인지
    last_index = [0,0]
    
    # 멀티탭에서
    for i in tap:
    	# 멀티탭에 있고 앞으로 쓸 제품에도 있다면
        # 가장 나중에 쓸(멀리 있는) 제품을 찾기위해
        # index를 비교해서 index가 큰 값으로 갱신
        if i in new and new.index(i) > last_index[0]:
            last_index[0] = new.index(i)
            last_index[1] = i
        # 멀티탭에 있는게 앞으로 쓸 제품에 없다면
        # 그거를 뽑는게 가장 효율적
        if i not in new:
            last_index[0] = 0
            last_index[1] = i
            break
    # 멀티탭에서 가장 나중에 쓸 제품을 제거
    tap.remove(last_index[1])
    # 멀티탭에 새로운 가전제품 꼽아주기
    tap.append(new.popleft())
    # 한번 뽑았으니까 count 1 증가
    count += 1
    
    # 사용할 가전제품이 남아있고 바로 쓸 물건이 멀티탭에 꼽혀있다면
    # 제거해주기
    while new and new[0] in tap:
        new.popleft()

print(count)

✏️ Comment

그리디를 사용한 문제이고 가장 핵심은 멀티탭에서 가장 나중에 쓸 제품을 골라서 빼는거라고 생각한다.