[TIL_Carrotww] 106 - 23/03/30

📝Carrotww의 코딩 기록장

🧲 python algorithm

🔍 programmers - 광물캐기 Level2

그리디 문제이다 다이아가 많은 순서대로 광물을 캐주면 된다
나는 다이아 철 돌에 가중치를 둬서 어떤걸 먼저 캘지 정해주었는데 가중치에 따라서 풀리는 테스트케이스가 있고 안되는 케이스가 있다.
문제를 다시 깔끔하게 풀어서 올려야겠다.

• 내 풀이

def solution(picks, minerals):
    result = 0

    minerals = minerals[:sum(picks)*5]
    slice_min = []

    def count_mineral(min_list):
        result = 0
        check_dict = {"diamond" : 15, "iron" : 5, "stone" : 1}
        for val in min_list:
            result += check_dict[val]

        return result

    cnt = 0
    tmp = []
    for mine in minerals:
        tmp.append(mine)
        cnt += 1
        if cnt == 5:
            weight = count_mineral(tmp)
            slice_min.append([weight, tmp])
            tmp = []
            cnt = 0

    if tmp:
        weight = count_mineral(tmp)
        slice_min.append([weight, tmp])

    slice_min.sort(key=lambda x:x[0])

    iron_graph = {"diamond" : 5, "iron" : 1, "stone" : 1}
    stone_graph = {"diamond" : 25, "iron" : 5, "stone" : 1}

    for i in range(3):
        for x in range(picks[i]):
            if slice_min:
                if i == 0:
                    mineral_list = slice_min.pop()[1]
                    result += len(mineral_list)
                elif i == 1:
                    mineral_list = slice_min.pop()[1]
                    for mineral in mineral_list:
                        result += iron_graph[mineral]
                else:
                    mineral_list = slice_min.pop()[1]
                    for mineral in mineral_list:
                        result += stone_graph[mineral]
            else:
                return result

    return result

첫 번째 풀이인데 더 깔끔하게 다듬어서 다시 올려야겠다.
그리디 문제는 더 나은 방법이 없을까 하면서 시간이 좀 걸리는것같다.
사실 쉬운 그리디 문제는 브루트포스랑 체감상 별 다를게 없기는 한데... 더 나은 방법이 없을까 하는 곳에 시간을 투자하다보니 코딩테스트에 그리디 문제가 나오면 좀 짜증난다 ㅠ
아무튼 요즘 velog 올리는걸 매번 올리지 않고 선택적으로 올리는데 다시 집중해서 맨날 올려야겠다~~!!