454. 四數相加 II
阿新 • • 發佈:2020-08-15
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
思路:先將 A+B 的和儲存在 map 中 (a+b, times),然後計算 C+D。
程式碼:
class Solution { public: int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) { unordered_map<int, int> hash_AB; int a, b, c, d, count = 0, sum; for(const auto &a:A) { for(const auto&b:B) { hash_AB[a+b]++; } } for(const auto &c:C) { for(const auto &d:D) { sum= -(c+d); if(hash_AB[sum]!=0) { count += hash_AB[sum]; } } } return count; } };