백준 - 로프(Swift)

백준 - 로프(Swift)

 

 

 문제 설명

 

https://www.acmicpc.net/problem/2217

2217번: 로프

 

 나의 풀이

 

이 문제는 로프가 3개 있을때 1개만 사용했을때의 최대값 vs 2개를 사용했을때의 최대값 vs 3개를 사용했을때의 최대값 중에 최대값을 구하면 된다.

 

1개만 사용했을때의 최대값은 가장 큰 값이 될 것이고 2개를 사용했을때의 최대값은 그 다음 가장 큰 값에 2를 곱하면 된다. 

이 때문에 정렬하고 반대로 탐색하면서 곱해주었다.

 

처음에는 배열에서 가장 작은 값을 찾고 인덱스를 곱해주고 지우는 방법으로 했는데 시간초과가 나서 아래 방식으로 했다.

 

import Foundation

let n = Int(readLine()!)!
var arr: [Int] = []
var max_val = 0
for _ in 0..<n {
    arr.append(Int(readLine()!)!)
}

arr.sort(by: >)

for i in stride(from: n-1, to: -1, by: -1) {
    max_val = max(max_val, arr[i]*(i+1))
}

print(max_val)