백준 1700번: 멀티탭 스케줄링 (파이썬)

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

1700번: 멀티탭 스케줄링

 

문제

 

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

 

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

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

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

 

입력

 

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

 

출력

 

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

 

풀이

 

전기 용품을 사용하려고 할 때, 크게 네 가지 경우가 있다.

첫 번째, 멀티탭이 가득 차 있으면서 이번에 사용하려고 하는 전기 용품이 있을 때다.

두 번째, 멀티탭이 가득 차 있으면서 이번에 사용하려고 하는 전기 용품이 없을 때다.

세 번째, 멀티탭이 가득 차 있지 않으면서 이번에 사용하려고 하는 전기 용품이 있을 때다.

네 번째, 멀티탭이 가득 차 있지 않으면서 이번에 사용하려고 하는 전기 용품이 없을 때다.

 

첫 번째 경우는 아무 일도 일어나지 않는다.

두 번째 경우는 멀티탭에서 하나의 전기 용품을 뽑고, 멀티탭에 새로운 전기 용품을 꽂아야 한다. 이때 무엇을 뽑을지 정해야 한다.

세 번째 경우는 아무 일도 일어나지 않는다.

네 번째 경우는 멀티탭에 새로운 전기 용품을 꽂아야 한다.

 

첫 번째 경우와 세 번째 경우는 pass로 처리해 주었다.

두 번째 경우에서 무엇을 뽑을지 결정할 때, 앞으로 가장 오랫동안 사용되지 않을 전기 용품을 뽑기로 했다. 이를 위해 리스트의 index 함수를 사용하였으며, 앞으로 사용하지 않거나 현재 순서로부터 사용 순서가 가장 멀리 떨어져 있는 전기 용품을 victim으로 정했다.

네 번째 경우는 멀티탭에 해당 전기 용품을 꽂아주었다.

 

 

소스 코드

 

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

s = list(map(int, input().split()))
m = []

cnt = 0
for i in range(k):
    if len(m) == n:
        if s[i] in m:
            pass
        else:
            if m[0] not in s[i:]:
                victim_i = 0
            else:
                max_d = s[i:].index(m[0])
                victim_i = 0
                for j in range(n):
                    if m[j] not in s[i:]:
                        victim_i = j
                        break
                    d = s[i:].index(m[j])
                    if max_d < d:
                        max_d = d
                        victim_i = j
            m[victim_i] = s[i]
            cnt += 1
    else:
        if s[i] in m:
            pass
        else:
            m.append(s[i])
print(cnt)