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

문제

1700번: 멀티탭 스케줄링

- 기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다.

- 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다.

- 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.

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

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

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

- 멀티탭 구멍의 개수 N (1 ≤ N ≤ 100)과 전기 용품의 총 사용 횟수 K (1 ≤ K ≤ 100)가 정수로 주어진다.

- 전기용품의 이름이 K 이하의 자연수로 사용 순서대로 주어진다.

알고리즘

- 멀티탭 구멍의 개수와 전기 용품의 총 사용 횟수를 입력받는다.

- 전기용품의 사용 순서대로 자연수로 입력받아 리스트 변수에 넣는다.

- 멀티탭에 플러그가 전기용품의 사용 순서대로 꽂혀있는지 확인한다. 

- 멀티탭에 플러그가 모두 꽂혀있으면 어떤 플러그를 먼저 뽑고 다음 플러그를 꽂아야 할지 선택한다.

- 멀티탭에 플러그를 꽂을 순서를 이미 알고 있기 때문에 그 순서를 인덱스를 통해 확인한다.

- 인덱스의 값이 클수록 플러그를 꽂을 순서가 뒤에 있는 것이기 때문에 인덱스의 값이 큰 것부터 뽑아준다.

- 플러그를 뽑을 때마다 카운트해준다. 

코드

import sys

n, k = map(int, sys.stdin.readline().split())
plug = list(map(int, sys.stdin.readline().split()))

# 멀티탭에 꽂혀있는 플러그
multitap = []
# 횟수
cnt = 0

for i in range(k):
    # 꽂아야하는 플러그가 멀티탭에 이미 꽂혀있으면 다음 loop를 돈다.
    if plug[i] in multitap:
        continue

    # 멀티탭에 1구라도 비어 있으면 멀티탭을 꽂고 다음 loop를 돈다.
    if len(multitap) < n:
        multitap.append(plug[i])
        continue

    # 다음에 꽂아야하는 플러그
    plug_next = []

    # 멀티탭에 모든 구를 사용하고 있을 때
    for j in range(n):

        # 멀티탭에 꽂혀 있는 플러그가 다음에 꽂아야하는 플러그중에 있는지 확인
        if multitap[j] in plug[i:]:
            # 다음에 꽂아야하는 플러그의 순서를 인덱스를 통해 확인하고 인덱스의 위치 값을 가져온다.
            # 인덱스 값을 다음에 꽂아야하는 플러그 리스트에 추가한다.
            plug_next.append(plug[i:].index(multitap[j]))

        else:
            # 멀티탭에 꽂혀 있는 플러그가 다음에 꽂아야하는 플러그중에 없으면 플러그 사용순서의 최대 범위 +1을 인덱스 값에 넣는다.
            # 인덱스 값인 101을 다음에 꽂아야하는 플러그 리스트에 추가하여 플러그를 최우선으로 뽑는다.
            plug_next.append(101)

            # 확실히 뽑아야하는 플러그가 생겼으므로 멈춰준다.
            break

    # 플러그를 인덱스 값이 큰 순서부터 뽑는다.
    del multitap[plug_next.index(max(plug_next))]
    # 다음 플러그를 꽂는다.
    multitap.append(plug[i])
    # 카운트
    cnt += 1

print(cnt)

결과

코드를 분석하게 되면 멀티탭에 전기용품 사용 순서대로 플러그가 꽂혀 있는지 확인을 하고 플러그가 모두 꽂혀있으면 꽂혀 있는 플러그중에 다음에도 사용해야 하는 플러그가 있는지 사용순서를 인데스를 통해 확인한다. 인덱스가 클수록 플러그를 뒤에 사용해도 되는 것이기 때문에 인덱스가 큰 순서부터 플러그를 뽑고 다음 플러그를 꽂으면 된다. 이때, 다음에도 사용해야 하는 플러그가 없다면 플러기 사용순서의 최대 범위 +1을 해주어 최우선으로 뽑게 해 준다.

문제를 풀면서 플러그를 인덱스 값이 큰 순서부터 뽑으려 할 때 remove() 함수를 통해 코드를 multitap.remove(plug_next.index(max(plug_next))) 라고 작성하여 인덱스가 제일 큰 값을 제거하려 했지만 Value Error가 뜨는 것을 확인할 수 있었다. 이유는 우리가 지우고자 하는 값이 아닌 인덱스 값으로 리스트를 제거하려고 했기 때문에 인덱스를 통해 제거하는 del 키워드를 통해 수행해야 하는 것을 확인할 수 있었다.

github

junjange/CodingTest