Leetcode 18 四數之和

2021-10-10 14:14:18 字數 2332 閱讀 6953

給定乙個包含 n 個整數的陣列 nums 和乙個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出所有滿足條件且不重複的四元組。

注意:答案中不可以包含重複的四元組。

示例:給定陣列 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

滿足要求的四元組集合為:

[[-1, 0, 0, 1],

[-2, -1, 1, 2],

[-2, 0, 0, 2]

]雙指標法

兩層遍歷,對剩下的元素建立左右指標。比較遍歷元素與左右指標之和是否與target相等。若相等,加入結果列表。

剪枝遍歷的第一層,若前四個元素之和大於target,break

遍歷的第一層,若遍歷的元素和列表最後3個元素之和小於target,continue

遍歷的第一層,若與之前元素相等,continue

遍歷的第二層,若確定的第一層元素與第二層的前3個元素之和大於target,break

遍歷的第二層,若確定的第一層元素與第二層元素與列表最後2個元素之和小於target,continue

遍歷的第二層,若與之前元素相等,continue

時間複雜度 o(n

3n^3

n3), 空間複雜度 o(1)

class

solution

:def

foursum

(self, nums: list[

int]

, target:

int)

-> list[list[

int]]:

nums =

sorted

(nums)

result =

for i in

range

(len

(nums)-3

):if nums[i]

+ nums[i +1]

+ nums[i +2]

+ nums[i +3]

> target:

break

if nums[i]

+ nums[

len(nums)-1

]+ nums[

len(nums)-2

]+ nums[

len(nums)-3

]< target:

continue

if i >

0and nums[i]

== nums[i -1]

:continue

for j in

range

(i +1,

len(nums)-2

):if nums[j]

+ nums[i]

+ nums[j +1]

+ nums[j +2]

> target:

break

if nums[j]

+ nums[i]

+ nums[

len(nums)-1

]+ nums[

len(nums)-2

]< target:

continue

if j > i +

1and nums[j]

== nums[j -1]

:continue

l, r = j +1,

len(nums)-1

while l < r:

mid_result = nums[i]

+ nums[j]

+ nums[l]

+ nums[r]

if mid_result == target:

[nums[i]

, nums[j]

, nums[l]

, nums[r]])

while l +

1< r and nums[l]

== nums[l +1]

: l +=

1while l +

1< r and nums[r]

== nums[r -1]

: r -=

1 l +=

1 r -=

1elif mid_result < target:

l +=

1else

: r -=

1return result

LeetCode 18 四數之和

給定乙個包含 n 個整數的陣列 nums 和乙個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d 使得 a b c d 的值與 target 相等?找出所有滿足條件且不重複的四元組。注意 答案中不可以包含重複的四元組。示例 給定陣列 nums 1,0,1,0,2,2 和 ...

LeetCode 18 四數之和

給定乙個包含 n 個整數的陣列nums和乙個目標值target,判斷nums中是否存在四個元素 a,b,c 和 d 使得 a b c d 的值與target相等?找出所有滿足條件且不重複的四元組。注意 答案中不可以包含重複的四元組。示例 給定陣列 nums 1,0,1,0,2,2 和 target ...

leetcode18 四數之和

def foursum nums,target numlen,res,d len nums set if numlen 4 return nums.sort 二層迴圈遍歷任意兩個元素對和存放到字典d裡並把序號存起來 for p in range numlen for q in range p 1,n...