[Python] 백준 1781: 컵라면

1781번: 컵라면 (acmicpc.net)

1781번: 컵라면

 

1. 컵라면수가 많은순으로 정렬한다.

2. find 함수로 날짜에 맞춰 채워나간다.

3. 날짜에 채웠으면 그 날짜에는 채우지 못하므로 배열의 값을 한개 내려준다.

4. find 함수로 계속해서 채울수있는 날짜를 채워나간다.

n = int(input())
ans = 0
pay = []
deadline = [i for i in range(200001)]


def find(x):
    if x == 0:
        return 0
    if x == deadline[x]:
        deadline[x] -= 1
        return x
    else:
        deadline[x] = find(deadline[x])
        return deadline[x]


for _ in range(n):
    pay.append(list(map(int, input().split())))

pay.sort(key=lambda x: -x[1])

for x, y in pay:
    if find(x) > 0:
        ans += y

print(ans)