454 四數相加 II

給定四個包含整數的陣列列表 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 。所有整數的範圍在 -228 到 228 - 1 之間，最終結果不會超過 231 - 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

方法一：分組 + 雜湊表

思路與演算法

我們可以將四個陣列分成兩部分，aa 和 bb 為一組，cc 和 dd 為另外一組。

對於 aa 和 bb，我們使用二重迴圈對它們進行遍歷，得到所有 a[i]+b[j]a[i]+b[j] 的值並存入雜湊對映中。對於雜湊對映中的每個鍵值對，每個鍵表示一種 a[i]+b[j]a[i]+b[j]，對應的值為 a[i]+b[j]a[i]+b[j] 出現的次數。

對於 cc 和 dd，我們同樣使用二重迴圈對它們進行遍歷。當遍歷到 c[k]+d[l]c[k]+d[l] 時，如果 -(c[k]+d[l])−(c[k]+d[l]) 出現在雜湊對映中，那麼將 -(c[k]+d[l])−(c[k]+d[l]) 對應的值累加進答案中。

最終即可得到滿足 a[i]+b[j]+c[k]+d[l]=0a[i]+b[j]+c[k]+d[l]=0 的四元組數目。

class solution 
}int ans = 0;
for (int u: c) }}
return ans;
}};