[백준-11399][그리디] ATM python

문제

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

11399번: ATM

 

풀이1

n = int(input())
people = list(map(int, input().split()))
# 오름차순정렬
# 예 : 3 1 4 3 2 -> 1 2 3 3 4
people.sort()

time = 0
# 누적 합 계산
# 1 / 1 2 / 1 2 3 / 1 2 3 3 / 1 2 3 3 4
for i in range(n):
    for j in range(0,i+1):
        time += people[j]
    
print(time)

풀이2(라이브러리)

n = int(input())
people = list(map(int, input().split()))
# 오름차순정렬
# 예 : 3 1 4 3 2 -> 1 2 3 3 4
people.sort()

time = 0

cum = np.cumsum(people)
# 누적 합 계산
for i in cum:
        time += i
    
print(time)