[Softeer - level2] 금고털이

 

 

import sys
input = sys.stdin.readline

# 배낭 무게 제한 W , n종류의 귀금속
# 가장 높은 가격으로 배낭 채우기 

w,n = map(int,input().split())
total_price = 0

#리스트 컴프리헨션
arr = [list(map(int,input().split())) for _ in range(n)]

arr.sort(key = lambda x:x[1],reverse=True)

idx = 0
while w>0:
    if arr[idx][0] >= w:
        total_price += arr[idx][1]*w
        break
    else:
        total_price += arr[idx][0]*arr[idx][1]
        w -= arr[idx][0]
        idx += 1 

print(total_price)

 

처음에 풀때는 리스트를 생성해두고 

입력값을 append 하였는데 시간초과가 계속 나왔다.

for _ in range(n):
    a,b = map(int,input().split())
    arr.append((b,a)) # b 무게 오름차순 정렬

 

 

 

 

List Comprehension의 속도가 빠른이유

TIL.77 List Comprehension의 속도가 빠른이유