6-pu2rile

[pu2rile]

🔗 문제 링크

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

✔️ 소요된 시간

20분

✨ 수도 코드

1. 문제 요약

• 사람은 1번부터 N번까지 번호가 매겨져 있으며, i번 사람이 돈을 인출하는데 걸리는 시간은 Pi분
• ex. 총 5명이 있고, P1 = 3, P2 = 1, P3 = 4, P4 = 3, P5 = 2인 경우 3번 사람은 1번, 2번 사람이 돈을 뽑을 때까지 기다려야 하기 때문에 총 3+1+4 = 8분이 필요
• 이 경우에 각 사람이 돈을 인출하는데 필요한 시간의 합은 3+4+8+11+13 = 39분
• 줄을 서 있는 사람의 수 N과 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어졌을 때, 각 사람이 돈을 인출하는데 필요한 시간의 합의 최솟값을 구하라

2. 알고리즘

• n 번째 사람이 인출할 때 n 번째 사람 앞까지 모든 사람의 시간을 time에 더해 주고 최소로 나오는 result를 출력
• P1 = 3, P2 = 1, P3 = 4, P4 = 3, P5 = 2인 경우, 시간이 적게 걸리는 시간 순으로 정렬하여 [P2, P5, P1, P4, P3] 순으로 누적합을 구하면 됨

🖥️ 코드

num = int(input())
arr = list(map(int, input().split())) 
arr.sort() #입력받은 정수 오름차순 정렬
result = 0
time = 0
for i in range(num): 
    time += arr[i] #현재 사람이 기다리는 시간을 계산해서 time에 더함
    result += time #현재 사람까지의 총 대기 시간을 결과에 더함
print(result)

📚 새롭게 알게된 내용

[oesnuj]

저도 풀어봤는데 똑같은 알고리즘으로 풀었네요
코드 잘 봤습니다!! 많이 늦었지만 PR 수고하셨어요~

[suhyun113]

이 문제는 그리디 유형이라 그런지 문제를 이해하는 것은 쉬웠어요! 합의 최솟값을 구해야하기 때문에 입력받은 값을 오름차순으로 정렬한다고 생각했어요! n번째 사람이 인출할 때 걸리는 시간은 이전의 사람이 걸리는 시간까지 합하여 계산하기 때문에 합의 최솟값을 구하기 위해서는 앞의 사람이 걸리는 시간이 적을 수록 유리해 처음에 오름차순으로 정렬이 필요합니당. 나머지는 수학적인 개념이 들어간 것 같아요!

저도 파이썬으로 구현해봤는데, 아래 코드 참고해주세용~

p = int(input()) # ATM에 줄을 선 사람 수
pi = list(map(int, input().split())) # 각 순서의 사람들이 돈을 인출하는데 걸리는 시간

pi.sort() # 돈을 인출하는데 걸리는 시간이 적은 순으로 정렬

current_sum = 0 # 현재 사람까지의 돈을 인출하는데 걸리는 시간의 합
total_waiting_time = 0 # 전체 돈을 인출하는데 걸리는 시간의 합

for time in pi: 
    current_sum += time
    total_waiting_time += current_sum

print(total_waiting_time)

신기한게,,,도현님이랑 완전 똑같이 짜졌어요ㅎㅎ 준서님도 같은 알고리즘으로 풀었다던데

따로 알아봤는데, 좀 더 수학적으로 접근해볼 수 있더라구요.

for i, p in enumerate(pi):
    total_waiting_time += p * (p-i)

enumerate()함수도 예전에 배웠던 기억이 있는데 잘 기억나지 않더라구요. 알아보니 인자로 넘어온 목록을 기준으로 인덱스와 원소를 차례로 접근해주는 반복자 객체를 반환해주는 함수였어요! 알고계실지도 모르지만 참고해도 좋을 것 같아요

두 방법 모두 시간 복잡도가 같아 어느 것이 더 효율적이다라고 말하기는 어렵지만,, 이런 방법도 있어서 신기했습니당
대신 수학적 풀이다 보니 처음에 구현한 코드보다 이해는 어려웠어요. 그래도 함수도 공부하게 되었고 이런 풀이도 있다는 것을 알게되어 유익했습니다ㅏ