[백준/Python] 2217 - 로프

고운·2024년 3월 31일

알고리즘

목록 보기
73/94

문제

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

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

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

입력

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

출력

첫째 줄에 답을 출력한다.


풀이
로프를 병렬로 연결해 중량을 w/k로 나누게 되더라도 가장 적은 중량을 들 수 있는 로프가 중요하게 된다
w/k의 값이 가장 적은 중량을 들 수 있는 로프의 중량보다 크다면 해당 로프는 사용할 수 없다
모든 로프를 사용해야한다면 단순히 허용 중량이 가장 적은 로프에 로프 개수를 곱하면 되겠지만, 모든 로프를 사용하지 않아도 되므로 로프들을 오름차순으로 정렬해놓고 최대 중량을 찾아나간다

예를 들어 [10 20 30 40 50] 순으로 로프들이 정렬되어있다면, n=5이므로 10의 중량을 갖는 로프는 중량 50까지는 견딜 수 있다
만약 첫번째 로프를 제외하면 20 ×\times 4 = 80 만큼의 중량을 버틸 수 있다
이런식으로 로프의 개수를 줄여나가며 최대 중량을 갱신하면 된다

코드

import sys

n = int(sys.stdin.readline())
rope = [int(sys.stdin.readline()) for _ in range(n)]

rope.sort()

answer = 0
for idx, elem in enumerate(rope):
    answer = max(answer, (n-idx)*elem)
print(answer)
profile
무럭무럭 성장하는 개린이 공부 공간

0개의 댓글