알고리즘 16일차

문제 출처: https://leetcode.com/problems/find-k-pairs-with-smallest-sums/submissions/

테스트 케이스는 통과하지만 submit 시 index out of range가 발생해 수정이 필요합니다.

func kSmallestPairs(_ nums1: [Int], _ nums2: [Int], _ k: Int) -> [[Int]] {

    var arr = [(Int, Int, Int)]()
    
    for i in nums1 {
        for j in nums2 {
            arr.append(((i+j), i, j))
        }
    }

    var temp = [Int]()
        
    for i in 0..<arr.count {
        temp.append(arr[i].0)
    }
        
    temp.sort(by: < )
    
    var res = [(Int, Int, Int)]()
        
    for i in temp {
        for j in arr {
            if i == j.0 {
                res.append((i, j.1, j.2))
            }
        }
    }
        
    var result = [[Int]]()
        
    for i in 0..<k {
        result.append([res[i].1, res[i].2])
    }
        
    return result
    
}