LeetCode 454 四數相加 II

2021-10-10 21:54:48 字數 1182 閱讀 8253

給定四個包含整數的陣列列表 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 和 b 為一組,c 和 d 為另外一組。

最終即可得到滿足 a[i]+b[j]+c[k]+d[l]=0 的四元組數目。

c++

class solution 

}int ans = 0;

for (int u: c) }}

return ans;

}};

python

class solution:

def foursumcount(self, a: list[int], b: list[int], c: list[int], d: list[int]) -> int:

countab = collections.counter(u + v for u in a for v in b) # 計數器

ans = 0

for u in c:

for v in d:

if -u - v in countab:

ans += countab[-u - v]

return ans

複雜度分析

1、

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