프로그래밍/Python
[Python] 백준 11399 ATM
모영이
2021. 2. 15. 18:18
11399번: ATM
첫째 줄에 사람의 수 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어진다. (1 ≤ Pi ≤ 1,000)
www.acmicpc.net
그리디 알고리즘, 다이나믹 프로그래밍
1) dp[i]는 atm[0]부터 atm[i]까지의 합을 저장한다.
2) dp[0]에 atm[0]을 넣어주고 for문을 시작한다.
3) dp의 합을 출력한다.
전체 코드
import sys
input = sys.stdin.readline
N = int(input())
atm = [0]*N
atm = list(map(int, input().split()))
atm.sort()
dp = [0]*N
dp[0] = atm[0]
for i in range(1, N):
dp[i] = dp[i-1] + atm[i]
print(sum(dp))