[백준] 1758 알바생 강호 - Python

문제

https://www.acmicpc.net/problem/1758

1758번: 알바생 강호

위 문제는 강호가 받을 수 있는 팁의 최댓값을 구해주어야 합니다.

팁은 (각 손님이 원래 생각했던 금액) - (받은 등수 -1)로 계산할 수 있으므로, 팁의 최댓값을 구하기 위해서는 주어지는 각 손님의 원래 손님이 생각했던 금액을 내림차순 정렬해주어야 합니다.

이를 통해 팁을 앞부터 순차적으로 구해주면 팁의 최댓값을 구해줄 수 있습니다.

Python

import sys
input = sys.stdin.readline
n = int(input())
people = []
for i in range(n):
    people.append(int(input()))

people.sort(reverse=True)
tip = 0

for idx,val in enumerate(people):
    if val - ((idx+1)-1) < 0:
        continue
    tip += val - ((idx+1)-1)
print(tip)