BAEKJOON #2217 로프 (Greedy) - python

nathan·2021년 6월 28일
0

알고리즘문제

목록 보기
2/102

로프

출처 : 백준 #2217

시간 제한메모리 제한
2초192 MB

문제

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

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

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


입력

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


출력

첫째 줄에 답을 출력한다.


입출력 예시

예제 입력 1

2
10
15

예제 출력 1

20


풀이

생각

  • 우선 역순으로 각 로프들을 정렬한다.

  • 각 로프들의 인덱스 + 1 만큼이 로프 개수를 의미하므로 자기 자신(로프)에 로프개수를 곱해준 만큼이 나눠 들 수 있는 최대 용량이라고 생각하면 된다. -> greedy

  • 들 수 있는 용량이 가장 큰 값을 선택한다.

풀이 설명

  • 큰 값부터 정렬이 되어야 인덱스 + 1(=로프 개수)를 적용할 수 있으므로, 우선 .sort(reverse=True)를 통해 역순으로 정렬해주었다.

  • 마지막에 리스트에서 최대 값을 뽑아내면 된다.

python code

# 백준 2217번
from sys import stdin

def rope(n, weights):
    weights.sort(reverse=True)
    for i in range(1, n):       
        weights[i] = weights[i]*(i+1)      # 누적 합으로 만들기 

    # print("weights", weights)
    return max(weights)



n = int(stdin.readline())
weights = []
for i in range(n):
    weights.append(int(stdin.readline()))


result = rope(n, weights)
print(result)
profile
나는 날마다 모든 면에서 점점 더 나아지고 있다.

0개의 댓글