우당탕탕 개발일지

[그리디] 백준 2217번 로프(Python 파이썬) 본문

알고리즘

[그리디] 백준 2217번 로프(Python 파이썬)

민아당긴아 2023. 11. 1. 14:24

문제

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

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

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

입력

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

출력

첫째 줄에 답을 출력한다.

예제 입력 1 복사

2
10
15

예제 출력 1 복사

20

해결방법

처음에는 문제 자체가 이해가 잘 안 되었다. 그래서 다른 블로그 글들을 보며 참고했다.

로프가 N개 있고, 이 중에서 몇 개의 로프를 선택해서 중량을 들지는 선택할 수 있다.

하지만 선택하는 로프들 중 가장 숫자가 작은 로프가 각 로프가 들 수 있는 최대 중량이다.

예를 들어, 로프가 3, 5, 9, 10, 13이라면(N=5인 상황)

선택한 로프의 개수 선택된 로프 각 로프가 들 수 있는 최대 중량 결과적으로 로프들이 들 수 있는 중량
1 13 13 13*1=13
2 10, 13 10 10*2=20
3 9, 10, 13 9 9*3=27
4 5, 9, 10, 13 5 5*4=20
5 3, 5, 9, 10, 13 3 3*5=15

결과적으로 로프들이 들 수 있는 중량 13, 20, 27, 20, 15 중에서 최대값은 27이므로 27을 출력하면 된다.

플로우

1. N값 입력받기 - input()

2. 각 로프가 버틸 수 있는 최대 중량 입력받기 - for(), input(), list

3. 로프의 개수별로 각 로프가 들 수 있는 최대 중량 계산하기 - for(), min()

4. 로프의 개수별로 로프들이 들 수 있는 최대 중량 계산하기 - min값*(로프의 개수)

5. 로프의 개수별로 로프들이 들 수 있는 최대 중량 중에서 최대값 구하기 - max()

6. 최대값 출력하기

코드

# 1. 
n = int(input())
# 2. 
rope = [int(input()) for _ in range(n)]
rope.sort(reverse=True)
# 3.
max_w = []
for i in range(1, n+1):
  m = rope[:i]
  # 4.
  max_w.append(i*min(m))

# 5. 
max_f = max(max_w)
print(max_f)