728x90
https://school.programmers.co.kr/learn/courses/30/lessons/172927
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
분석
- 사용할 수 있는 곡괭이중 아무거나 하나를 선택해 광물을 캡니다.
- 한 번 사용하기 시작한 곡괭이는 사용할 수 없을 때까지 사용합니다.
- 광물은 주어진 순서대로만 캘 수 있습니다.
- 광산에 있는 모든 광물을 캐거나, 더 사용할 곡괭이가 없을 때까지 광물을 캡니다.
picks [1, 3, 2] ([dia, iron, stone]과 같은 구조)
minerals ["diamond", "diamond", "diamond", "iron", "iron", "diamond", "iron", "stone"]
minerals diamond diamond diamond iron iron diamond iron stone picks dia dia dia dia dia iron iron iron picks 사용 1 2 3 4 5 1 2 3 피로도 1 1 1 1 1 5 1 1
- 광물 5개씩 끊어 리스트 생성 : 곡괭이 하나당 5개의 광물을 캐기 때문
- 5광물 리스트를 sort : 다이아, 철, 돌 순으로 정렬하여 피로도를 낮게 캐기 위해
- 5광물 리스트를 돌면서 곡괭이 사용
- 곡괭이 사용 잔여 횟수 감소
- 피로도 추가
picks [0, 1, 1] ([dia, iron, stone]과 같은 구조)
minerals ["diamond", "diamond", "diamond", "diamond", "diamond", "iron", "iron", "iron", "iron", "iron", "diamond"]
광물 5개씩 끊은 리스트 : [[5, 0, 0], [0, 5, 0], [0, 0, 0]]
곡괭이 선택
picks, j, picks[j]
[0, 1, 1] 0 0 다이아 곡괭이 없음
[0, 1, 1] 1 1 철 곡괭이로 광물 캐기
[0, 0, 1] 0 0 다이아 곡괭이 없음
[0, 0, 1] 1 0 철 곡괭이 없음
[0, 0, 1] 2 1 돌 곡괭이로 광물 캐기
[0, 0, 0] 0 0 다이아 곡괭이 없음
[0, 0, 0] 1 0 철 곡괭이 없음
[0, 0, 0] 2 0 돌 곡괭이로 광물 캐기
풀이
def solution(picks, minerals):
answer = 0
if len(minerals)>sum(picks)*5:
minerals=minerals[:sum(picks)*5]
# 광물 5개씩 끊은 리스트
mlist=[[0,0,0] for _ in range((len(minerals))//5+1)]
for i in range(len(minerals)):
if minerals[i]=='diamond':
mlist[i//5][0]+=1
elif minerals[i]=='iron':
mlist[i//5][1]+=1
elif minerals[i]=='stone':
mlist[i//5][2]+=1
mlist.sort(key=lambda x:(-x[0], -x[1], -x[2]))
for x in mlist:
dia,iron,stone = x
for j in range(len(picks)):
if picks[j]>0 and j==0: # 다이아몬드 곡괭이
picks[j]-=1
answer+=dia+iron+stone
break
elif picks[j]>0 and j==1: # 철 곡괭이
picks[j]-=1
answer+=dia*5+iron+stone
break
elif picks[j]>0 and j==2: # 돌 곡괭이
picks[j]-=1
answer+=dia*25+iron*5+stone
break
return answer
728x90