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

 

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

길이가 같은 배열 A, B 두개가 있습니다. 각 배열은 자연수로 이루어져 있습니다. 배열 A, B에서 각각 한 개의 숫자를 뽑아 두 수를 곱합니다. 이러한 과정을 배열의 길이만큼 반복하며, 두 수를 곱

programmers.co.kr

 

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

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

먼저 최솟값을 구하려면 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
}
  • 네이버 블러그 공유하기
  • 네이버 밴드에 공유하기
  • 페이스북 공유하기
  • 카카오스토리 공유하기
// custom