[BOJ]1700. 멀티탭스케줄링 (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차원 배열로 저장했다.
2. 새로운 플러그가 안꽃혀 있는 경우, 기존 플러그 중 다음 사용될 순서가 가장 큰 것을 뽑는다! (당연히, 다음에 사용될 일이 없는 플러그는 1순위이다.)
3. 새로운 플러그의 번호와 다음에 사용될 순서를 각각 추가해준다.
4. 새로 꽃을 플러그가 이미 꽃혀 있는 플러그라면 해당 플러그의 다음에 사용될 순서만 최신화 해준다!

 

 

코드

 

 

import sys
read = sys.stdin.readline

tab_N, using_K = map(int, read().strip().split())

elec_items = list(map(int, read().strip().split()))


# 0번 인덱스에는 현재 꽃힌 item, 1번 인덱스에는 그 item이 또 등장하는 다음 index를 담았다.
# 공간복잡도가 비효율적인 것 같다. (추측)
now_items = [[], []]

# 플러그를 빼는 횟수 카운트
count = 0
for i in range(len(elec_items)):
    item = elec_items[i]
    # 안꽃힌 플러그를 꽃아야 하는 경우, now_items[1]의 가장 작은 인덱스를 빼주고, 새로운 플러그의 값, 그 플러그의 다음 값을 기록한다. count를 하나 올린다.
    if item not in now_items[0]:
        if len(now_items[0]) == tab_N:
            try:
                remove_idx = now_items[1].index(-1)
            except:
                remove_idx = now_items[1].index(max(now_items[1]))
            del now_items[0][remove_idx]
            del now_items[1][remove_idx]

            try:
                temp = elec_items.index(item, i + 1)
            except:
                temp = -1
            now_items[0].append(item)
            now_items[1].append(temp)
            count += 1
        else:
            now_items[0].append(item)
            try:
                temp_idx = elec_items.index(item, i + 1)
            except:
                temp_idx = -1
            now_items[1].append(temp_idx)
    # 이미 꽃힌 플러그의 경우, count를 올리지 않고 해당 플러그의 다음 올 index만 최신화 해준다.
    else:
        temp_idx = now_items[0].index(item)
        try:
            now_items[1][temp_idx] = elec_items.index(item, i + 1)
        except:
            now_items[1][temp_idx] = -1


print(count)