[백준] 11399번 ATM 파이썬

오랜만에 다시 알고리즘 풀기 시작!

 

11399번 ATM 풀이

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

numbers.sort() #정렬
pre = 0 #이전까지의 합
result = 0 #총 합계
for i in range(0, n):
    result += pre + numbers[i] #이전합에 현재값을 더하여 총합계 만들기
    pre += numbers[i] #다음 차례에 쓰일 이전 값 갱신
print(result)

사람의 수와 각 사람이 돈을 인출하는 데 걸리는 시간을 입력으로 받는다.

5 그리고 3 1 4 3 2 를 입력으로 하면, 5명의 사람이 차례로 3분, 1분, 4분, 3분, 2분의 시간 동안 돈을 인출하는데 두 번째 사람부터는 앞사람이 돈을 인출하는 동안 대기해야 한다. 결과적으로 각 사람이 돈을 인출하는데 필요한 시간의 최소 합을 구하는 것이 문제이다.

예시대로라면 인출 순서에 따라 3 + (3+1) + (3+1+4) + (3+1+4+3) + (3+1+4+3+2) = 39의 값이 될 수도 있고, 1 + (1+2) + (1+2+3) + (1+2+3+3) + (1+2+3+3+4) = 32도 될 수 있다. 예시에서는 32가 최솟값이 된다. 앞 순서에 인출하는 사람의 소요 시간이 계속해서 누적으로 더해지기 때문에 인출 소요 시간이 짧은 사람이 앞 순서에 위치하도록 해야 한다. 그러므로 sort()를 사용해서 정렬을 해주고 누적합을 계산한다. 

처음에는 입력받은 사람들을 순열로 배치하고, 모든 경우의 합을 계산하여 최솟값을 구해야 한다고 생각해서 어떻게 순열을 만들지 고민했었다. 하지만 앞 순서 사람의 소요 시간을 최소로 만들어야 한다는 것을 깨닫고 정렬을 통해 단순히 문제를 해결할 수 있었다.