개발공부/코테

[백준 2217번: 로프] 파이썬 풀이

sohee! 2023. 8. 13. 12:43
 

2217번: 로프

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

www.acmicpc.net

문제

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

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

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

입력

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

출력

첫째 줄에 답을 출력한다.


접근 방법

ex) 10 15일 때,

10 10 인 20이 최대 (15*1 < 10*2)

 

ex) 9 10 15일 때

15 보다 9 9 9가 더 크니까 최대 (15*1 < 10*2 < 9*3)

 

큰 순서대로 정렬 후 → 15*1 < 10*2 < 9*3 순서대로 뭐가 최대인지 구하기

 

풀이

N = int(input()) # 로프의 개수 
rope_list = [] # rope_list라는 리스트에 각 로프의 최대 중량을 저장
result_list = [] # 비교 결과를 담을 리스트

for _ in range(N): # for문을 로프의 개수(N)만큼 돌려서 입력을 받은 후 
  rope_list.append(int(input()))  # 리스트에 저장
  
rope_list.sort(reverse=True) # 리스트를 정렬 (오름차순)

cnt = 1
for i in range(N):
  result_list.append(rope_list[i]*cnt)
  cnt += 1

print(max(result_list))
  • 로프의 개수 N을 받아서 for문을 로프의 개수(N)만큼 돌려서 입력을 받은 후 rope_list라는 리스트에 저장
  • rope_list를 정렬 (sort) → 오름차순으로
  • 첫 번째 수 1 , 두 번째 수2, 세 번째 수 *3 …. 숫자를 result_list에 넣어서
  • max(result_list)로 최대를 구함!