[Python][백준] 1700. 멀티탭 스케줄링 / 그리디 (G1)

🔗링크 :  

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

---

---

🗒️파이썬 코드 풀이

N,K = map(int,input().split())
lst = list(map(int,input().split()))
cnt = 0

multitab = set()
for i in range(len(lst)):
  
  # 장치가 멀티탭에 꽂힌 경우  
  if lst[i] in multitab: 
    continue 

  # 멀티탭 자리가 남는 경우
  if len(multitab) < N : 
    multitab.add(lst[i])
    continue

  # 멀티탭 자리 부족으로 장치 교체 
  last_used = {}
  for device in multitab:
    if device in lst[i+1:]:
      last_used[device] = lst[i+1:].index(device)
    else:
      last_used[device] = float('inf')

  remove_device = max(last_used,key=last_used.get)
  multitab.remove(remove_device)
  multitab.add(lst[i])
  cnt += 1 

print(cnt)

 

1. 세가지의 조건으로 나눈다.

• device가 멀티탭에 꽂힌 경우
• 멀티탭 자리가 남는 경우
• 멀티탭 자리 부족으로 device 교체

2.  "멀티탭 자리 부족으로 device 교체"  이 부분이 중요한데, 

멀티탭의 값이 lst의 i+1 ~ N 안에 있는지를 확인한다. 

 

3. 값이 있으면 해당 lst의 인덱스를 찾아 last_used 딕셔너리 넣고 

 

4. 값이 없으면 inf를 last_used 딕셔너리에 넣어줘서 제거 1순위로 만들어준다. 

 

5. 그리고 max 함수로 제거 할 device를 정하고 새로 꽂을 lst[i] 를 넣어준다. 

(각 순간에 최적이라 되는 선택 → 그리디 )

 

📌 문제 코멘트

1. 멀티탭 자리가 남는 경우를 아래와 같이 생각했다.

if i < N :
    multitab.add(lst[i])
    continue

어차피 device가 remove 되자마다 add 될 것이기 때문에 위와 같이 작성해도 문제가 없을 줄 알았다. 

하지만 전기용품이  N 전에 중복적으로 사용될 수 있기때문에 위와 같이 하면 안되고 len(multitab)으로 해야한다. 

ex) 1,1,2,3  (사용 순서)  → 이런 경우 i < N 으로 하면 문제 발생

 

2.  또 실수 했던 부분은, lst[i+1 : ] 의 부분이다.

맨 처음 코드를 작성했을 때, lst[i+1 : i+N]만 신경쓰면 되는 줄 알았다.

하지만 단지 i+N 범위만 신경쓰는게 아니라, 남은 lst 전부를 신경써야한다. 

→ 다시 사용의 가능성이 있기 때문에 !

---

📚문제

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

예를 들어 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