[Algorithm] [Python&Swift] 백준/BOJ - 2217 _ 로프

2217 - 로프

 

문제

---

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

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

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

 

입력

---

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

 

출력

---

첫째 줄에 답을 출력한다.

 

제출코드

---

Python

n = int(input())
arr = sorted([int(input()) for _ in range(n)], reverse=True)
result = 0
for i in range(n):
    if result < arr[i] * (i+1):
        result = arr[i] * (i+1)
print(result)

 

Swift

func main() -> Int {
    let n = Int(readLine()!)!
    var result = 0
    var arr = [Int]()
    for _ in 0..<n {
        arr.append(Int(readLine()!)!)
    }
    arr.sort(by: >)
    for i in 0..<n {
        if result < arr[i] * (i+1) {
            result = arr[i] * (i+1)
        }
    }
    return result
}
print(main())

 

문제풀이

---

k개의 로프를 이용하여 w 무게의 물체를 들 때, 각 로프에는 w/k의 중량이 걸린다.

가지고 있는 로프들을 모두 사용할 필요가 없기에 로프 최대 하중무게를 내림차순으로 정렬한 뒤 모든 경우의 수를 찾아 비교한다.

예를 들어, 입력값이 [40, 80, 60, 20 ,100] 인 경우 이를 [100, 80, 60, 40, 20] 으로 정렬한 뒤 비교를 진행한다.

0번째 로프만 쓸 경우, 100kg

0,1번째 로프를 쓸 경우, 160kg

0,1,2번째 로프를 쓸 경우, 180kg

이와 같은 방식으로 순차적으로 접근하여 최대 하중무게를 찾아낸다.