454 四數相加 II(Python)

2021-10-10 21:52:43 字數 1469 閱讀 5480

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

解釋:兩個元組如下:

(0, 0, 0, 1) -> a[0] + b[0] + c[0] + d[1] = 1 + (-2) + (-1) + 2 = 0

(1, 1, 0, 0) -> a[1] + b[1] + c[0] + d[0] = 2 + (-1) + (-1) + 0 = 0

class

solution

:def

foursumcount

(self, a: list[

int]

, b: list[

int]

, c: list[

int]

, d: list[

int])-

>

int:

n=len(a)

res=

0for i in

range

(n):

for j in

range

(n):

for k in

range

(n):

for l in

range

(n):

if a[i]

+b[j]

+c[k]

+d[l]==0

: res+=

1return res

兩兩相加,然後存在字典裡面查詢;

注意重複值也算,因為是不同資料加出來的,而不同資料是對應不同index。

class

solution

:def

foursumcount

(self, a, b, c, d)

: res =

0 dic =

for a in a:

for b in b:

total1 = a + b

dic[total1]

= dic.get(total1,0)

+1for c in c:

for d in d:

total2 = c + d

if-total2 in dic:

res += dic[

-total2]

return res

454 四數相加 Leetcode

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

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

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 例如 輸入...