454. 4 Sum II

454. 4 Sum II

這一個題目的設計比較特別一點，給出四個長度一樣的陣列，要從四個陣列中挑出任意四個數字，其總和為零。

這個題目存在著暴力解，那就是四個陣列一個一個掃描，這樣的時間複雜度會是 $O(n^4)$ 如果說每個陣列的長度為 n 。

不過這一個題目如果用 n Sum 的思想來想的話，其實存在著另外一個解法，那就是如果我從前面兩個陣列，各挑出一個數字後，其總和如果是 k ，那我只要再另外兩個陣列，找出總和為 -k 的組合即可。

有一件事情要注意的是，從前面兩個陣列隨機挑選兩的個數字，其總和為 k 的情況其實有很多種，假設有 x 種，那後面兩個陣列每次找出總和為 -k 的情況，就可以組合出 x 種組合。

因此在計算答案時，要記得累加可以組合的情況。

這​樣的方式時間複雜度可以降低到只要在 $O(n^2)$。

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        target = 0
        table = defaultdict(int)
        ans = 0
        for num1 in nums1:
            for num2 in nums2:
                s = num1 + num2
                table[s] = table[s] + 1

        for num3 in nums3:
            for num4 in nums4:
                s = num3 + num4
                if target - s in table:
                    ans += table[target-s]
        return ans