【一天一大 lee】四數相加 II (難度:中等) - Day20201127

題目:

給定四個包含整數的陣列列表  A , B , C , D ,計算有多少個元組 (i, j, k, l) ，使得  A[i] + B[j] + C[k] + D[l] = 0。

為了使問題簡單化，所有的 A, B, C, D 具有相同的長度  N，且 0 ≤ N ≤ 500 。所有整數的範圍在 $-2^{2}8$ 到 $2^{2}8$ - 1 之間，最終結果不會超過  $2^{3}1$ - 1 。

示例:

輸入:
A = [ 1, 2]
B = [-2,-1]
C = [-1, 2]
D = [ 0, 2]

輸出:
2

解釋:
兩個元組如下:
1. (0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0

拋磚引玉

暴力迴圈 || 分組迴圈 -> 超時

/**
 * @param {number[]} A
 * @param {number[]} B
 * @param {number[]} C
 * @param {number[]} D
 * @return {number}
 */
var fourSumCount = function(A, B, C, D) {
    let N = A.length,
        _result = 0,
        listA = [],
        listB = []
    // A、B先成組匹配
    for (let a = 0; a < N; a++) {
        for (let b = 0; b < N; b++) listA.push(A[a] + B[b])
    }
    // C、D再成組匹配
    for (let c = 0; c < N; c++) {
        for (let d = 0; d < N; d++) listB.push(C[c] + D[d])
    }
    // 組合兩個組
    for (let c = 0; c < listA.length; c++) {
        for (let d = 0; d < listB.length; d++) {
            if (listA[c] + listB[d] === 0) _result++
        }
    }
    return _result
}

上面邏輯超時，那麼優化下分組的邏輯，在分組時，組內相同結果會在組合兩個組時重複計算，那麼使用雜湊來去重

/**
 * @param {number[]} A
 * @param {number[]} B
 * @param {number[]} C
 * @param {number[]} D
 * @return {number}
 */
var fourSumCount = function(A, B, C, D) {
    let N = A.length,
        _result = 0,
        map = new Map()
    for (let a = 0; a < N; a++) {
        for (let b = 0; b < N; b++) {
            map.set(A[a] + B[b], (map.get(A[a] + B[b]) || 0) + 1)
        }
    }
    for (let c = 0; c < N; c++) {
        for (let d = 0; d < N; d++) {
            if (map.has(-C[c] - D[d])) {
                _result += map.get(-C[c] - D[d])
            }
        }
    }
    return _result
}

部落格: 前端小書童

每天的每日一題，寫的題解會同步更新到公眾號一天一大 lee 欄目
歡迎關注留言

公眾號：前端小書童