[swea] 5201. 컨테이너 운반 / python 파이썬

🚩 그리디

 

thinking

화물의 용량이 트럭의 적재용량보다 작으면 결과값에 더해주는 문제인데,

화물의 개수(N)가 더 많거나 트럭의 개수(M)가 더 많을 경우

인덱스에러(list index out of range)가 나기 때문에 while 문으로 처리하여,

N과 M의 최소 크기보다 작을 때만 반복문이 돌아가도록 설정했다.

 

코드

T = int(input())


for tc in range(1, T+1):
    N, M = map(int, input().split())  # N: 컨테이너수, M: 트럭수
    weight = sorted(list(map(int, input().split())), reverse=True)  # N개의 화물 무게
    truck = sorted(list(map(int, input().split())), reverse=True)  # M개 트럭의 적재용량

    res = 0
    cnt = 0
    i, j = 0, 0  # truck, weight
    while True:
        if cnt == min(N, M):
            break
        if truck[i] >= weight[j]:
            res += weight[j]
            i += 1
            j += 1
        else:
            j += 1
        cnt += 1

    print("#{} {}".format(tc, res))