PS

BOJ - 2217. 로프 (파이썬)

kugnuoy 2023. 9. 18. 15:44

2217.

문제

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

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

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

입력

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

출력

첫째 줄에 답을 출력한다.

 

[해설]

로프의 개수 num을 입력받는다.

로프의 각 최대 무게를 내림차순으로 정렬하여 저장한다.

 

해당 로프들로 만들 수 있는 무게 조합은 항상 num개 이다.

 

(1) num=1인 경우,

무게 조합은 rope[0] 하나만 존재한다.

(2) num=2인 경우,

무게 조합은 1개사용: rope[0] x 1  (rope[0]만 사용하는 경우)

                    2개사용: rope[1] x 2  (rope[0]과 rope[1]을 사용하는 경우)

두 가지 존재한다. 

(3) num=3인 경우,

무게 조합은 1개사용: rope[0] x 1  (rope[0]만 사용하는 경우)

                    2개사용: rope[1] x 2  (rope[0]과 rope[1]을 사용하는 경우)

                    3개사용: rope[2] x 3  (rope[0]과 rope[1]과 rope[2]를 사용하는 경우)

세 가지 존재한다. 

 

무게 조합 중에 최댓값을 출력하면 된다.

[풀이]

num = int(input())    
rope = []

# rope weight 입력
for i in range(num):
    tmp = int(input())    
    rope.append(tmp)

# rope weight 정렬
rope = sorted(rope, reverse=True)   

# 만들 수 있는 무게를 answer에 저장
answer = []

for i in range(num):
    answer.append(rope[i]*(i+1))

print(max(answer))