题目描述

)

代码实现

import UIKit

var nums1 = [1,2]
var nums2 = [-2, -1]
var nums3 = [-1, 2]
var nums4 = [0, 2]

func fourSumCount(_ nums1: [Int], _ nums2: [Int], _ nums3: [Int], _ nums4: [Int]) -> Int {
    var cnt = 0
    
    var nums5 = [Int : Int]()
    
    for num1 in nums1 {
        for num2 in nums2 {
            nums5[num1 + num2] = (nums5[num1 + num2] ?? 0) + 1
        }
    }
    
    for num3 in nums3 {
        for num4 in nums4 {
            if nums5.keys.contains(0 - num3 - num4) {
                cnt += nums5[0 - num3 - num4] ?? 0
            }
        }
    }

    return cnt
}

fourSumCount(nums1, nums2, nums3, nums4)

Q.E.D.