https://www.acmicpc.net/problem/1700
성능 요약
메모리: 113112 KB, 시간: 124 ms
분류
그리디 알고리즘
문제 설명
기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.
예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면,
- 키보드
- 헤어드라이기
- 핸드폰 충전기
- 디지털 카메라 충전기
- 키보드
- 헤어드라이기
키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다.
입력
첫 줄에는 멀티탭 구멍의 개수 N (1 ≤ N ≤ 100)과 전기 용품의 총 사용횟수 K (1 ≤ K ≤ 100)가 정수로 주어진다. 두 번째 줄에는 전기용품의 이름이 K 이하의 자연수로 사용 순서대로 주어진다. 각 줄의 모든 정수 사이는 공백문자로 구분되어 있다.
출력
하나씩 플러그를 빼는 최소의 횟수를 출력하시오.
아이디어, 문제풀이
- 아이디어 대신, chat gpt로 만든 주석들을 첨부한다.
for i in range(k): # n_list를 순회합니다.
if n_list[i] in set_list: # 만약 현재 기기가 이미 멀티탭에 꽂혀 있다면, 건너뜁니다.
continue
if len(set_list) < n: # 만약 멀티탭에 여유 공간이 있다면, 현재 기기를 꽂습니다.
set_list.append(n_list[i])
else: # 멀티탭에 여유 공간이 없다면, 교체할 기기를 찾아야 합니다.
max_use = 0 # 다음 사용까지 걸리는 시간을 나타냅니다.
idx_to_remove = 0 # 교체할 기기의 인덱스입니다.
for j in range(len(set_list)): # 멀티탭에 꽂혀있는 각 기기에 대해
next_use = 0 # 해당 기기가 다음에 사용될 때까지의 시간을 나타냅니다.
for future_item in n_list[i+1:]: # 현재 기기 이후에 사용되는 기기들을 순회합니다.
if set_list[j] == future_item: # 현재 기기가 멀티탭에 있는 기기와 같다면, 멈춥니다.
break
next_use += 1 # 다음 사용까지 걸리는 시간을 증가시킵니다.
if next_use > max_use: # 다음 사용까지 걸리는 시간이 현재 최대보다 크다면
max_use = next_use # 최대 시간을 업데이트하고
idx_to_remove = j # 교체할 기기의 인덱스를 업데이트합니다.
set_list[idx_to_remove] = n_list[i] # 멀티탭에서 교체할 기기를 제거하고, 새 기기를 꽂습니다.
cnt += 1 # 교체 횟수를 증가시킵니다.TROUBLE SHOOTING
- 아래는 처음에 문제를 풀이할때 접근했던 방법이다.
단순히 그 코드가 그 이후에 사용이 된다면 뽑지않고, 그게 아니라면 다 뽑는다… 라는식으로 구현했는데, 이전에 아이디어를 확장하며 도착한 지점이라 그전에도 수많은 고민을 했던 것 같다. 생각보다 어려웠다.. 일단 내 머릿속에 있는 아이디어를 코드로 구현을 해야했는데, 그 방법이 떠오르질 않았다. 그래서 chat gpt와 함께 코드를 풀었다. 처음에는 인덱스를 따로 선언해서, 그인덱스에 맞춰 저장하고 빼내는 방법을 추천해줬는데, 오늘안에 이해할 수 있을까 생각해봤을때, 꽤 오랜시간이 걸릴것 같아 다른 방법을 찾아 해맸다.def fun(): set_list = [] cnt = 0 for i in range(k): visit = [0] * k # if n_list[i] in set_list: # continue if len(set_list) < n: set_list.append(n_list[i]) # continue # if n_list[i] in set_list: # continue else: if n_list[i] in set_list: continue else: for j in range(i, k): if n_list[j] in set_list: visit[j] = 1 else: set_list.remove print(visit) return cnt
- 결론적으로, 인덱스를 사용하지만, 최대한 내가 이해할 수 있는 방향으로 코드를 짰다. 코드는 어려워보이는데, 사용할 수 있는 변수들과 배열들을 지정해주고, 계속 초기화를 해주며 각 상황에 맞게 쓴게 전부다. 다른 쉬운방법도 있을테고… 다른 알고리즘을 사용해 구현한 코드도 있을텐데, 나는 지금 이 코드가 제일 이해하기 쉬운것 같다. 언젠가 한단계 더 성장해서 다른 방법들도 쉽게 이해할 수 있어야 할텐데… 지금은 깊게 생각하지 않는게 좋을것 같다.
코드
#https://www.acmicpc.net/problem/1700
#멀티탭 스케줄링
#1700
import sys
input = sys.stdin.readline
n, k = map(int, input().split())
n_list = list(map(int, input().split()))
def fun():
set_list = []
cnt = 0
for i in range(k):
if n_list[i] in set_list:
continue
if len(set_list) < n:
set_list.append(n_list[i])
else:
max_use = 0
index = 0
for j in range(len(set_list)):
next_use = 0
for future_item in n_list[i+1:]:
if set_list[j] == future_item:
break
next_use += 1
if next_use > max_use:
max_use = next_use
index= j
set_list[index] = n_list[i]
cnt += 1
print(set_list)
return cnt
answer = fun()
print(answer)
해보자! 게임 클라 개발자!