Leetcode 四數相加 II

2021-09-12 02:06:19 字數 1834 閱讀 5072

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

這裡直接放官方的解題思路,不過這些情況我也都考慮到了,並同樣採用了第三種方式,可惜還是沒有官方的方法簡潔,有興趣的可以直接看官方的**。

暴力解法,時間複雜度 o(n^4)

將 d 中的元素放入查詢表,時間複雜度 o(n^3)

將 a + b 的每一種可能放入查詢表,然後 兩重迴圈對在查詢表中尋找是否存在 -(c[i] + d[i]),時間複雜度為 o(n^2)。兩數相加的和作為鍵值。

我採用第三種方法實現

class solution(object):

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

"""dic_1 = {}

dic_2 = {}

for idx, num_1 in enumerate(a):

for idy, num_2 in enumerate(b):

result = num_1 + num_2

if result in dic_1:

_list = dic_1.get(result)

dic_1[result] = _list

else:

_list = [(idx, idy)]

dic_1[result] = _list

for idx, num_1 in enumerate(c):

for idy, num_2 in enumerate(d):

result = num_1 + num_2

if result in dic_2:

_list = dic_2.get(result)

dic_2[result] = _list

else:

_list = [(idx, idy)]

dic_2[result] = _list

result = 0

for number_1 in dic_1.keys():

target = 0 - number_1

if target in dic_2:

list_1 = dic_1.get(number_1)

list_2 = dic_2.get(target)

result += len(list_1) * len(list_2)

return result

1. 

LeetCode 四數相加 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,...

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