PGS / 광물 캐기 / Python

PGS / 광물 캐기

풀면서

✅ 광물의 순서를 바꿀 수는 없으나 한번 채굴 시작 시 무조건 5개는 채굴해야 한다. 즉, 5개로 묶은 chunk들에 대한 재정렬은 가능하다.

✅ 전체 광물을 5개의 단위로 자르고, 각 청크 내에서 다이아 -> 철 -> 돌 의 순서로 점수 차를 둬서 이를 기준으로 청크들을 정렬한다.

✅ 정렬된 청크들에 대해서는 단순히 다이아 -> 철 -> 돌 곡괭이 순서로 채굴을 진행하면 최소 피로도가 누적된다.
</br>

해결 코드1

def solution(picks, minerals):
    answer = 0
    end = min(sum(picks)*5, len(minerals))
    
    minerals_to_num = []
    # 광물들 5개 단위로 자르기
    for i in range(0, end, 5):
        if i+5 > end:
            minerals_ = minerals[i:]
        else:
            minerals_ = minerals[i:i+5]
        d, i, s = minerals_.count("diamond"), minerals_.count("iron"), minerals_.count("stone")
        minerals_to_num.append([d, i, s])
        
    minerals_to_num.sort(key = lambda x : (-x[0], -x[1]))
    
    for m in (minerals_to_num):
        if picks[0] > 0:
            picks[0] -= 1
            answer += sum(m)
        elif picks[1] > 0:
            picks[1] -= 1
            answer += (m[0]*5 + m[1]*1 + m[2]*1)
        else:
            picks[2] -= 1
            answer += (m[0]*25 + m[1]*5 + m[2]*1)
            
    return answer