프로그래머스 - 최솟값 만들기 [Swift]

https://programmers.co.kr/learn/courses/30/lessons/12941?language=swift 

코딩테스트 연습 - 최솟값 만들기

 

법칙만 찾아내면 간단한 문제.

그 법칙을 알려줄테니 여러분도 법칙만 보고 풀어보시길.

먼저 최솟값을 구하려면 A의 요소 중 가장 작은 것과 B의 요소 중 가장 큰 것을 곱하고 더하면 된다.

예를 들어 A: [2, 4, 5], B: [3,4,1] 이라고 한다면

가장 작은 값은 (2 * 4) + (4 * 3) + (5 * 1) 이 된다는 것이다.

(어이없게 쉽죠?)

 

정답코드: 

import Foundation

func solution(_ A:[Int], _ B:[Int]) -> Int
{
    var ans = 0
    var n = A.count
    
    var A = A.sorted(by: { $0 < $1 }) // min~max
    var B = B.sorted(by: { $1 < $0 }) // max~min
    
    for i in 0..<n {
        ans += A[i] * B[i]
    }
    return ans
}