[백준] 11399번 파이썬 (ATM)

1. 문제와 예제

(문제 링크: https://www.acmicpc.net/problem/11399)

11399번: ATM

문제

예제

---

2. 전체 코드

# 입력 받기
N = int(input().rstrip()) # 줄 서 있는 사람 수
times = list(map(int, input().split())) # 각 사람이 돈 인출하는 데 걸리는 시간 리스트
res = 0 # 필요한 시간의 합 최솟값

times.sort() # 정렬

for i in range(N):
    res += times[i] * (N-i)

# 결과 출력하기
print(res)

---

3. 코드 해설

예제를 보면, times가 정렬 후에 [1, 2, 3, 3, 4]가 되고 시간의 합은 1 + (1+2) + (1+2+3) + (1+2+3+3) + (1+2+3+3+4)이다. 정리하면 1*5 + 2*(5-1) + 3*(5-2) + 4*(5-3) + 5*(5-4)이다. 이를 그대로 코드로 적용하면 된다.