[백준/파이썬] 11399 ATM - 그리디

소스 코드 1 128ms

import sys
n = int(sys.stdin.readline())

time_list = [int(x) for x in sys.stdin.readline().strip().split()]

time_list.sort() # 오름차순 정렬

answer = 0
for i in range(n):
    sum = 0
    for j in range(i+1):
        sum += time_list[j]
    answer += sum

print(answer)

소스 코드 2 72ms

import sys
n = int(sys.stdin.readline())

time_list = list(map(int, input().split()))

time_list.sort()

answer = 0
count = list(range(n))

for t, c in zip(time_list, count):
    answer += t*(n-c)
print(answer)

주요 개념

• 한 줄에 숫자 여러 개 입력 받아서 리스트로 만들기

time_list = [int(x) for x in sys.stdin.readline().strip().split()]

time_list = list(map(int, input().split()))