[백준 2217번] [🥈4] 로프 (python)

백준 사이트 링크

문제

N(1 ≤ N ≤ 100,000)개의 로프가 있다. 이 로프를 이용하여 이런 저런 물체를 들어올릴 수 있다. 각각의 로프는 그 굵기나 길이가 다르기 때문에 들 수 있는 물체의 중량이 서로 다를 수도 있다.

하지만 여러 개의 로프를 병렬로 연결하면 각각의 로프에 걸리는 중량을 나눌 수 있다. k개의 로프를 사용하여 중량이 w인 물체를 들어올릴 때, 각각의 로프에는 모두 고르게 w/k 만큼의 중량이 걸리게 된다.

각 로프들에 대한 정보가 주어졌을 때, 이 로프들을 이용하여 들어올릴 수 있는 물체의 최대 중량을 구해내는 프로그램을 작성하시오. 모든 로프를 사용해야 할 필요는 없으며, 임의로 몇 개의 로프를 골라서 사용해도 된다.

입력

첫째 줄에 정수 N이 주어진다. 다음 N개의 줄에는 각 로프가 버틸 수 있는 최대 중량이 주어진다. 이 값은 10,000을 넘지 않는 자연수이다.

출력

첫째 줄에 답을 출력한다.

예제 입력 1

2
10
15

예제 출력 1

20

---

문제를 풀면서 막혔던 부분

• 처음에는 가장 작은 무게를 들 수 있는 로프만을 기준으로 병렬 연결 로프 갯수를 곱했다. 여기서 출력 초과, 시간 초과가 나와 무거운 무게순으로 내림차순 정렬을 한 후에 로프가 들 수 있는 질량 * 병렬 연결 로프 갯수로 바꾸어 주었다.

예시)

$[100, 20, 10] \rightarrow [100 \times 1, 20 \times 2, 10 \times 3]$

---

코드

ver(1)

import sys
n = int(sys.stdin.readline())
k = []
result = []
for i in range(n):
  k.append(int(sys.stdin.readline()))
k.sort(reverse=True)
for num in range(n):
  result.append(k[num] * (num+1))
print(max(result))

ver(2) - def 사용

import sys
n = int(sys.stdin.readline())
k = []
result = []

def s(k):
  k.sort(reverse=True)
  for j in range(n):
    k[j] = k[j] * (j + 1)
  return max(k)



for i in range(n):
  k.append(int(sys.stdin.readline()))

print(s(k))