leetCode 454 四數相加

2021-10-10 21:52:43 字數 1579 閱讀 7569

目錄

一、題目描述

二、解題思路

三、**實現

給定四個包含整數的陣列列表 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...