leetcode 454 四數相加 雜湊表

2021-09-19 07:00:58 字數 1332 閱讀 1378

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

解題思路:可將原命題轉為求a[i]+b[j] = -(c[k]+d[l]),

1. 將a[i]+b[j]作為map的key存入,同時統計該key出現的次數,即遍歷a[i]+b[j]的所有組合,並統計各組合出現的次數。

2. 把-(c[k]+d[l])的所有組合作為key,並在map中查詢該key是否存在,同時累加各key出現的次數。

注意c[k] d[l])對應的所有組合中,二者之和即-(c[k]+d[l])可能會有很多是相同的,這不要緊,因為查詢到-(c[k]+d[l])後,我們一次一次累加其出現的次數

下面給出具體的**及後續優化

int leetcode_hash::foursumcounti(vector& a, 

vector& b, vector& c, vector& d)

} for (const auto& c : c)

} return result;

}

class solution 

}for (auto c : c) }}

return result;

}};

時間優化:主要對result += map[-(c+d)]進行優化,這句話會在map中遍歷key即-(c+d),如果該key不存在則將該key插入到map中,然後map還要自動對該key進行排序,浪費大量時間;這裡直接使用fing查詢該key是否存在,若存在則累加對應的value;否則進行下一次查詢

空間優化:指定map的大小;上述時間優化的過程也極大程度優化了空間的消耗

leetCode 454 四數相加

目錄 一 題目描述 二 解題思路 三 實現 給定四個包含整數的陣列列表 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 ...

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...