알고리즘/백준

백준 Python - 11399 ATM

Wonjun Sung 2021. 2. 8. 15:59

www.acmicpc.net/problem/11399

 

11399번: ATM

첫째 줄에 사람의 수 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어진다. (1 ≤ Pi ≤ 1,000)

www.acmicpc.net


코드:

n=int(input())
arr=list(map(int,input().split()))
arr.sort()
p=[]
for i in range(1,n+1):
    p.append(sum(arr[:i]))
print(sum(p))

 

풀이:

  1. 테스트 케이스 , 시간을 입력받고 arr은 최소 시간을 구하기 위해서 정렬시킨다.
  2. 개인별로 소요되는 시간 Pi를 담을 빈 리스트 p 를 생성한다.
  3. arr에는 시간 순으로 나열되어 있으므로 for 문으로 하나씩 늘려주면서 합을 구해주고 개인 소요 시간을 구한다.
  4. p 리스트의 합을 구하면 끝!