[알고리즘/백준] 2217번 : 로프(python)

작은거 * 전체 갯수가 답이라고 생각했다. 모든 경우의 수를 리스트에 넣고 제일 큰 값을 출력하면 답이라고 생각. 시간은 고려하지 않았다....

a = list(int(input()) for _ in range(int(input())))
m = []
for i in sorted(a):
    m.append(i*len(a))
    a.remove(i)
print(max(m))

그랬더니 시간초과가 나왔다......................

전에 값보다 더 크면 그 값만 가지고 있게 바꾸었다

a = list(int(input()) for _ in range(int(input())))
tmp = 0
for idx, k in enumerate(sorted(a, reverse=True)):
    if tmp < k * (idx+1):
        tmp = k * (idx+1)
print(tmp)