문제
기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.
예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면,
- 키보드
- 헤어드라이기
- 핸드폰 충전기
- 디지털 카메라 충전기
- 키보드
- 헤어드라이기
키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다.
입력
첫 줄에는 멀티탭 구멍의 개수 N (1 ≤ N ≤ 100)과 전기 용품의 총 사용횟수 K (1 ≤ K ≤ 100)가 정수로 주어진다. 두 번째 줄에는 전기용품의 이름이 K 이하의 자연수로 사용 순서대로 주어진다. 각 줄의 모든 정수 사이는 공백문자로 구분되어 있다.
출력
하나씩 플러그를 빼는 최소의 횟수를 출력하시오.
예제 입력 1
2 7
2 3 2 3 1 2 7
예제 출력 1
2
👧 나의 풀이
# 실패
from collections import Counter
import sys
input = sys.stdin.readline
N, K = map(int, input().split()) # 멀티탭 구멍 개수, 전기 용품 총 사용횟수
using = list(map(int, input().split()))
dic = Counter(using)
tap = [0] * N # 멀티탭
cnt=0
while using:
thing = using.pop(0)
for i in range(N):
if not tap[i] : # 빈 멀티탭 구명이 있다
tap[i] = thing
dic[thing] -= 1
break
else:
if thing in tap : # 꽂혀있으면
dic[thing]-=1
continue
# 플러그 빼고 다른 플러그 thing 꽂기
for j in range(N):
if not dic[tap[j]] : # 더이상 쓸일 없는 플러그 뽑기
tap[j] = thing
dic[thing] -= 1
cnt += 1
break
else : # 가장 나중에 쓸 것 뽑기
tap[-1] = thing
dic[thing] -= 1
cnt += 1
print(cnt)
dic에 제품별로 사용횟수를 딕셔너리로 만들었다.
tap에 꽂을 때마다 해당 제품의 값을 1 감소했다.
플러그를 뽑을 때 고려해야할 점
1. 더이상 쓸일이 없는 플러그를 우선으로 뽑는다.
2. 가장 나중에 사용할 제품을 뽑는다.
이 두 가지를 만족하도록 해야한다. 나는 1번만 만족시켜서 실패한듯 ㅠ
# 실패 : ValueError
from collections import Counter
import sys
input = sys.stdin.readline
N, K = map(int, input().split()) # 멀티탭 구멍 개수, 전기 용품 총 사용횟수
using = list(map(int, input().split()))
dic = Counter(using)
tap = [0] * N # 멀티탭
cnt=0
while using:
thing = using.pop(0)
for i in range(N):
if not tap[i] : # 빈 멀티탭 구명이 있다
tap[i] = thing
dic[thing] -= 1
break
else:
if thing in tap : # 꽂혀있으면
dic[thing]-=1
continue
# 플러그 빼고 다른 플러그 thing 꽂기
for j in range(N):
if not dic[tap[j]] : # 더이상 쓸일 없는 플러그 뽑기
tap[j] = thing
dic[thing] -= 1
cnt += 1
break
else :
#last=using.pop() # 마지막에 사용할 제품
x=tap.index(using[-1])
tap[x] = thing
dic[thing] -= 1
#dic[using[-1]]-=1
cnt += 1
break
print(cnt)
# 실패
import sys, heapq
input=sys.stdin.readline
N, K=map(int, input().split()) # 멀티탭 구멍 개수, 전기 용품 총 사용횟수
E=list(map(int, input().split())) # 전기 용품 사용 순서
tap=[0]*N
cnt=0
for k in range(K) :
for idx, t in enumerate(tap) :
if not t : # 아무 것도 안 꽂혀있을 때
tap[idx]=E[k]
break
else : # 다른 플러그를 뽑아야 함
if E[k] in tap :
continue
cnt += 1
change=0
i=0
for idx, t in enumerate(tap) :
if t not in E[k+1:] : # 더이상 안쓰면 뽑기
i=idx
break
elif change < E[k+1:].index(t) : # 꽂혀있는 것 중에서 최대한 나중에 쓸 것 뽑기
change=E[k+1:].index(t)
i=idx
tap[i]=E[k]
print(cnt)
🧒 다른 사람의 풀이
# https://pacific-ocean.tistory.com/357
n, k=map(int, input().split()) # 멀티탭 구멍의 개수, 전기 용품의 총 사용횟수
s=list(map(int, input().split())) # 전기용품들
m=[0 for i in range(n)] # 멀티탭
cnt=0
for i in range(k) :
isTrue=False
for j in range(n) :
if m[j] == s[i] or m[j] == 0 : # 멀티탭에 이미 꽂혀있거나 비어있거나
isTrue=True
m[j]=s[i]
break
if isTrue :
continue
else : # 플러그 뽑기
a=0
for j in range(n) :
try :
if a < s[i+1:].index(m[j]):
a=s[i+1:].index(m[j])
b=j
except:
a=-1
b=j
break
m[b]=s[i]
cnt+=1
print(cnt)
플러그를 뽑을 때 쓸 제품들s[i+1:]에서 멀티탭에 꽂혀있는 제품m[j]중 늦게 사용하는 제품을 뽑는다.
a는 멀티탭에서 m[j]의 위치
s[i+1:]에서 m[j]가 없으면 에러가 떠서 except문 실행=더이상 사용할 일이 없는 제품 플러그를 뽑는다.
#https://angangmoddi.tistory.com/165
from sys import stdin
N, K = stdin.readline().split()
N = int(N)
K = int(K)
multap = [0] * N
li = list(map(int, stdin.readline().split()))
res = swap = num = max_I = 0 # 플러그 뽑는 횟수, 뽑을 제품 플러그, li를 슬라이싱할 인덱스, swap의 인덱스
for i in li:
if i in multap: # 이미 꽂은 플러그면 넘어가기
pass
elif 0 in multap: # 빈 곳에 플러그 꽂기
multap[multap.index(0)] = i
else:
for j in multap:
if j not in li[num:] : # 더이상 j제품을 쓸일이 없다
swap = j
break
elif li[num:].index(j) > max_I: # 가장 나중에 사용할 전기용품을 뽑는다
max_I = li[num:].index(j)
swap = j
multap[multap.index(swap)] = i # 새로운 제품 플러그 꽂기
max_I = swap = 0
res += 1 # 플러그 뽑는 횟수
num += 1
print(res)
위랑 로직이 같음
문제 출처 👉 백준
'coding test' 카테고리의 다른 글
[파이썬] 1197. 최소 스패닝 트리 (0) | 2021.06.13 |
---|---|
[파이썬] 1916. 최소비용 구하기 (0) | 2021.06.11 |
[파이썬] 12851. 숨바꼭질 2 (1) | 2021.05.25 |
[파이썬] 16953. A -> B (0) | 2021.05.24 |
[파이썬] 2606. 바이러스 (0) | 2021.05.23 |