目錄
一、題目描述
二、解題思路
三、**實現
給定四個包含整數的陣列列表 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 。所有整數的範圍在
例如:
輸入: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
參考兩數之和的思想,假設求陣列中是否存在滿足arr[i] + arr[j] = sum的i和j,我們會把arr中的數放進雜湊表中,然後去查詢雜湊表中是否存在等於sum-arr[j]的key。
這樣一想這道題只不過把數放進四個陣列裡,每個陣列裡面找滿足加起來和為0的數字。所以把陣列合併,直到變成兩數之和的模型即可:
把a和b中的數兩兩相加,把和放進雜湊表中,表中記錄和出現的次數;
把c和d中數兩兩求和,求和過程中如果步驟1雜湊表中存在0 - c[i] - d[j],那麼說明找到了滿足要求的元祖,把0 - c[i] - d[j]出現的次數加進結果裡即可。
#includeusing namespace std;
int foursumcount(vector& a, vector& b, vector& c, vector& d)
} unordered_mapcdsum;
for (int i = 0; i < n; ++i)
} int res = 0;
for (auto ite = absum.begin(); ite != absum.end(); ++ite)
} return res;
}//稍微優化下:只開乙個雜湊表,第二表及結果在累加的時候就計算出來
int foursumcount2(vector& a, vector& b, vector& c, vector& d)
} int res = 0;
for (int i = 0; i < n; ++i)
} }return res;
}int main() ;
vectorb = ;
vectorc = ;
vectord = ;
cout << foursumcount2(a, b, c, d);
return 0;
}
leetcode454 四數相加 II
class solution def foursumcount self,a,b,c,d type a list int type b list int type c list int type d list int rtype int 如果暴力做法就會是o n 4 且 0 n 500 那麼6250...
leetcode454 四數相加 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...
LeetCode 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...