[BOJ: 2217] - Python / 파이썬 - 로프

o_jooon_·2024년 1월 22일
0

BOJ

목록 보기
8/49
post-thumbnail

서론

그리디 문제입니다.
기본적인 문제로, 모든 로프를 사용해야 할 필요는 없으며 이 부분이 핵심입니다.
예를 들어, n = 3, 로프 = 1, 5, 20인 경우,
로프 3개를 모두 사용하면 최대 무게는 3,
5와 20만 사용하면 10,
20만 사용하면 20이 되기 때문에 크기 순서대로 탐색을 해주어야 합니다.
때문에 정렬을 사용한 후 탐색해주면 됩니다.

난이도

실버 4


문제

조건


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

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

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


입력

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


출력

첫째 줄에 답을 출력한다.


예시

예제 입력 1

2
10
15

예제 출력 1

20

풀이

  1. 로프를 입력받는다.
  2. 로프를 내림차순으로 정렬한다. (뒤에서부터 빼주기 용이하게)
  3. 현재 로프의 수와 로프에서 가장 작은 무게를 견딜 수 있는 마지막 인덱스를 곱하여 최대 무게와 비교한다.
  4. 마지막 로프는 없애준다.
  5. 모두 탐색하면 종료한다.

코드

import sys
input = sys.stdin.readline

ropes = sorted([int(input()) for _ in range(int(input()))], reverse=True)
ans = 0

while ropes:	
	# 현재 있는 로프 수와 가장 마지막 로프의 무게를 확인해야 하기 떄문에
	# len과 pop으로 코드를 조금 더 짧게 줄였다.
    ans = max(ans, len(ropes) * ropes.pop())

print(ans)

실행 결과

profile
안녕하세요.

0개의 댓글