프로그래밍/문제풀이

[BOJ-1700] 멀티탭 스케줄링 Python 풀이

Cycrypt0 2023. 5. 6. 22:11
728x90

[Gold I] 멀티탭 스케줄링 - 1700

문제 링크

성능 요약

메모리: 34120 KB, 시간: 68 ms

분류

그리디 알고리즘

문제 설명

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

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

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

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

입력

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

출력

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

풀이

아이디어

콘센트의 개수만큼 0으로 초기화된 리스트를 하나 만들어주고 앞에서부터 리스트를 확인해 들어온다.
만약에 콘센트가 0이라면 무조건 일단 꼽고 본다.(단 아래 테스트 케이스처럼 비어있는 중에 같은 번호의 전열기구가 꼽힌다면 꼽지 않는다.)

3 11 
11 8 11 7 2 8 2 7 5 10 2

만약 전열기구가 모두 꼽혀있다면 두가지 조건을 볼 수 있는데

  • 앞으로 더이상 해당 전열기구가 나오지 않는경우 (우선순위 가장 높음)
  • 꼽힌 전열기구 중 가장 나중에 등장하는 전열기구

첫번째 조건이 만족하면 flag를 True로 만들어주고, 두번째 조건을 돌지 않는다
만약 두번째 조건이라면 index()를 이용하여 concent에 들어있는 전열기구 중 가장 나중에 등장하는 전열기구를 max() 함수를 이용하여 찾아주고, 콘센트에서 빼준 뒤 p로 업데이트 해준다.

코드

from collections import deque
tap, l = map(int, input().split())
lst = deque(list(map(int, input().split())))
concent = [0 for _ in range(tap)]    # 콘센트를 구의 수만큼 초기화해준다
cnt = 0
while lst:
    p = lst.popleft()
    if p in concent:    # 먼저 콘센트에 같은게 꼽혔는지 확인해주고 
        continue

    elif 0 in concent:
        for idx, connected in enumerate(concent):    # 안꼽힌데가 있으면 꼽아준다.
            if not connected:
                concent[idx] = p
                break
    ## 위 두 조건문의 순서를 바꿔서 했다가 "틀렸습니다"가 떴다.
    else:
        flag = False
        for x, i in enumerate(concent):
            if i not in lst:
                flag = True
                concent[x] = p
                cnt += 1
                break

        if not flag:
            id_lst = [lst.index(i) for i in concent]    
            # 좀 복잡한데, index를 통해 가장 먼 전열기구를 먼저 뽑아준다.
            concent[id_lst.index(max(id_lst))] = p
            cnt += 1

print(cnt)

회고

아이디어를 짜내고 반례를 잡아내는데 시간이 많이 들었다. 골드I 정도의 그리디를 자력으로 풀었다는 것에 뿌듯함과 자신감을 느낀다.

또한 질문게시판을 찾아보며 OPT-2알고리즘이라는 것에 대해 알게 되었는데, 새로 공부해 봐야 할 문제가 하나 추가되었다.

728x90