2022.03.05 [백준] (python 파이썬) ATM

 

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

11399번: ATM

 

---

풀이

문제에서 모든 사람이 돈을 인출할 때, 필요한 가장 최단 시간을 찾으라고 한다.

그러면 입력받은 배열을 오름차순 정렬하면 시간이 짧게 걸리는 순으로 정렬되고,

총 n명일 때, 가장 처음 인출한 사람의 시간이 n번 반복된다.

총 인원 5명, 걸리는 시간[1, 2, 3, 4, 5]로 예를 들면

1번 사람이 인출한 시간 : 1 = 1

2번 사람이 인출한 시간 : 1 2 = 3

3번 사람이 인출한 시간 : 1 2 3 = 6

4번 사람이 인출한 시간 : 1 2 3 4 = 10 

5번 사람이 인출한 시간 : 1 2 3 4 5 = 15

위의 예시처럼 1이 5번, 2가 4번, 3이 3번 4가 2번 5가 1번 반복되게 된다.

위의 방법을 코드로 옮기면 아래와 같다.

n=int(input())
arr=list(map(int, input().split()))
arr.sort()
ans=0
for i in arr:
    ans+=i*n
    n-=1
print(ans)