[백준 2217번 로프] 문제 해설 - 파이썬

[백준 2217번 로프] 문제 해설 - 파이썬

문제

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

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

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

입력

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

출력

첫째 줄에 답을 출력한다.

백준 링크

• 백준 링크

 

---

문제 접근법

• 로프를 병렬로 연결하면 각 로프에는 w/k만큼의 동일한 중량이 걸립니다.
• 즉, 가장 작은 무게를 들 수 있는 로프가 들 수 있는 질량 * 병렬 연결 로프 갯수 = 최종 무게
• 가장 무거운 무게를 들 수 있는 로프 부터 내림차순으로 정렬합니다.
• [100, 80, 60, 40, 20] 이 있다고 하면 [1001, 802, 603, 404, 20*5] 합니다.
• 여기서 가장 큰 값이 답입니다.

파이썬 코드

def solution():
    answer = 0
    arr.sort(reverse=True)
    for i in range(N):
        arr[i] = arr[i] * (i + 1)

    return max(arr)


N = int(input())
arr = []
for _ in range(N):
    arr.append(int(input()))

print(solution())