[BOJ] 11399번 ATM 파이썬 풀이

map함수를 사용시 첫번째 인자로 함수, 두번째 인자로 스플릿한 문자열을 넣어주었다.
map함수의 두번째 인자는 첫번째 인자인 함수의 인자로 각각 들어가게 된다.

 

sum_time += p[i] * (n-i) 

수식의 경우 순서대로 각 사람이 걸리는 시간을 뒤에 남은 사람들이 기다려야 하기 때문에 위와 같은 수식을 사용이 가능하다.

n = int(input())                   # 정수 n 입력받기
p = sorted(list(map(int, input().split())))  # n명의 사람이 돈을 인출하는데 걸리는 시간을 입력받아 정렬하여 리스트 p에 저장

sum_time = 0                       # 걸리는 총 시간을 0으로 초기화
for i in range(n):                 # 0부터 n-1까지 반복문 수행
    sum_time += p[i] * (n-i)       # p[i]에 걸리는 시간에, 그 사람을 기다리는 시간 (n-i)을 곱한 값을 sum_time에 더함

print(sum_time)                    # 총 걸린 시간을 출력함