백준) 11399번 ATM

min:D·2022년 7월 5일
0

알고리즘 공부

목록 보기
6/10
시간 제한메모리 제한
1초256MB

🏧 문제

인하 은행에는 ATM이 1대밖에 없다.
지금 이 ATM 앞에 N명의 사람들이 줄을 서있다. 사람은 1번 부터 N번까지 번호가 매겨져 있으며, i번 사람이 돈을 인출하는데 걸리는 시간은 PiP_i분이다.

사람들이 줄을 서는 순서에 따라서, 돈을 인출하는데 필요한 시간의 합이 달라지게 된다.
예를 들어, 총 5명이 있고, P1=3,P2=1,P3=4,P4=3,P5=2P_1=3, P_2=1, P_3=4, P_4=3, P_5=2인 경우를 생각해보자.
[1,2,3,4,5] 순서로 줄을 선다면,

  • 1번 사람 = 3분
  • 2번 사람 = 3 + 1 = 4분
  • 3번 사람 = 3 + 1 + 4 = 8분
  • 4번 사람 = 3 + 1 + 4 + 3 = 11분
  • 5번 사람 = 3 + 1 + 4 + 3 + 2 = 13분
  • 각 사람이 돈을 인출하는데 필요한 시간의 합은 3 + 4 + 8 + 11 + 13 = 39분

줄을 [2,5,1,4,3]의 순서로 선다면,

  • 2번 사람 = 1분
  • 5번 사람 = 1 + 2 = 3분
  • 1번 사람 = 1 + 2 + 3 = 6분
  • 4번 사람 = 1 + 2 + 3 + 3 = 9분
  • 3번 사람 = 1 + 2 + 3 + 3 + 4 = 13분
  • 총 1 + 3 + 6 + 9 + 13 = 32분

줄을 서 있는 사람의 수 N과 각 사람이 돈을 인출하는데 걸리는 시간 PiP_i가 주어졌을 때, 각 사람이 돈을 인출하는데 필요한 시간의 합의 최솟값을 구하는 프로그램을 작성하시오.

🔢 입력

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

📊 출력

첫째 줄에 각 사람이 돈을 인출하는데 필요한 시간의 합의 최솟값을 출력한다.

📑 예제 입력1 -> 출력1

5
3 1 4 3 2
32

💻 코드

n = int(input())
p = list(map(int, input().split()))

# 인출 시간이 가장 작을 순서로 정렬
p.sort()

tmp = []
for i in range(len(p)):
    tmp.append(sum(p[:i+1]))
    
print(sum(tmp))

0개의 댓글