[Swift/스위프트] 백준 1026번: 보물

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

1026번: 보물

 

소스코드

import Foundation

var n = Int(readLine()!)
var a = readLine()!.components(separatedBy: " ").map{Int($0)!}.sorted(by: <)
var b = readLine()!.components(separatedBy: " ").map{Int($0)!}.sorted(by: >)

var sum = 0
for i in 0..<n{
    sum+=a[i]*b[i]
}

print(sum)

 

풀이

readLine으로 읽은 string 배열을 int로 변환하기 위해 고차 함수 map을 이용하였다.

 

문제에서 입력받은 b는 재배열을 하면 안 된다고 하여 계속 고민하다 인터넷을 찾아보니 어차피 최솟값만 출력하는 것이기 때문에 두 배열 모두 정렬을 바꿔주어도 상관없다.라는 결론이 나왔다. 따라서 sorted함수를 이용해 각각 오름,내림 차순으로 정렬하여 계산하였다. 

// 오름차순
arr.sort()
arr.sort(by: <)
arr = arr.sorted(by: <)

// 내림차순
arr.sort(by: >)
arr = arr.sorted(by: >)