給定乙個包含 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...