문제풀이/BOJ

[Python] BOJ/백준 11399번 ATM

서채리 2021. 8. 23. 22:52

[문제]

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

 

11399번: ATM

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

www.acmicpc.net

 


[풀이]

처음 각 사람이 돈 인출에 걸리는 시간을 리스트에 넣을 때 sorted() 함수를 이용해 오름차순으로 리스트를 정렬해 저장한다.

 

 

temp와 res를 0으로 초기화한다.

temp에는 for문을 돌 때마다 위 첨부사진의 두 번째 줄 숫자가 저장된다.

res는 temp 값을 더한다. (3번째 줄)

 


[코드]

import sys

if __name__ == '__main__':
    n = int(sys.stdin.readline())
    line = sorted(list(map(int, sys.stdin.readline().split())))

    temp, res = 0, 0
    for i in line:
        temp += i
        res += temp
    print(res)