광물캐기

https://school.programmers.co.kr/learn/courses/30/lessons/172927

프로그래머스

 

분석

• 사용할 수 있는 곡괭이중 아무거나 하나를 선택해 광물을 캡니다.
• 한 번 사용하기 시작한 곡괭이는 사용할 수 없을 때까지 사용합니다.
• 광물은 주어진 순서대로만 캘 수 있습니다.
• 광산에 있는 모든 광물을 캐거나, 더 사용할 곡괭이가 없을 때까지 광물을 캡니다.

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

1. 광물 5개씩 끊어 리스트 생성 : 곡괭이 하나당 5개의 광물을 캐기 때문
2. 5광물 리스트를 sort : 다이아, 철, 돌 순으로 정렬하여 피로도를 낮게 캐기 위해
3. 5광물 리스트를 돌면서 곡괭이 사용
   1. 곡괭이 사용 잔여 횟수 감소
   2. 피로도 추가

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