[백준] 1700번 > 멀티탭 스케줄링

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

1700번: 멀티탭 스케줄링

문제

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

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

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

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

입력

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

출력

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

 

---

풀이

 

골드1 문제치고는 비교적 수월(?)하게 풀렸던 것 같다. 반례만 잘 찾는다면 ㅋㅋㅋ

 

먼저 멀티탭이 비어있다면 그냥 꼽으면 된다.

다 차있다면 사용할 전자제품이 멀티 탭에 이미 꼽혀있는지 검사한다.

          이미 꼽혀있다면 continue

 

꼽혀 있지 않다면 이제 어떤 전자제품의 플러그를 뽑을 것인가를 선택하여야한다. 그 기준은 플러그에 꼽혀있는 전자제품이 추후에 다시 사용되는가이다. 

          다시 사용된다면 pass

          다시 사용되지 않는다면 그 제품의 플러그를 뽑는다!!(break를 통해 빠져나옴)

 

여기서 또 하나의 case는 플러그에 꼽혀있는 모든 제품이 추후에 다시 사용되는 case이다.(이는 for-else문으로 판별가능)이때에는 index()메서드를 통해 가장 늦게 사용되는 제품의 플러그를 뽑는다.(Greedy 방식)

 

import sys
input = sys.stdin.readline
n, k = map(int, input().split())
elec_items = list(map(int, input().split()))
answer = 0
multi_tap = []
for i in range(len(elec_items)):
    if len(multi_tap) < n and elec_items[i] not in multi_tap:
        multi_tap.append(elec_items[i])
    else:
        if elec_items[i] in multi_tap:
            continue
        for j in multi_tap:
            if j in elec_items[i+1:]:
                pass
            else: # 해당 전자제품이 뒤에서 다시 사용되지 않는 case
                answer += 1
                multi_tap.remove(j)
                multi_tap.append(elec_items[i])
                break
        else: # 모든 전자제품이 뒤에서 다시 사용되는 case
            next_schedule_idx = []
            for socket in multi_tap:
                next_schedule_idx.append((elec_items[i+1:].index(socket), socket))
            multi_tap.remove(max(next_schedule_idx)[1])
            multi_tap.append(elec_items[i])
            answer += 1
print(answer)

코드가 좀 지저분한 것 같기도..;;