프로그래밍/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))

 

'프로그래밍 > Python' 카테고리의 다른 글

[Python] 백준 11404 플로이드  (0) 2021.02.16
[Python] 백준 2470 두 용액  (0) 2021.02.15
[Python] 백준 11047 동전0  (0) 2021.02.15
[Python] 백준 10773 제로  (0) 2021.02.15
[Python] 백준 2164 카드 2  (0) 2021.02.15