給定乙個包含 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]
]
前情提要:leetcode1.兩數之和,leetcode15. 三數之和,leetcode16. 最接近的三數之和
class
solution
:def
foursum
(self, nums, target)
:"""
:type nums: list[int]
:type target: int
:rtype: list[list[int]]
"""res, length =
,len
(nums)
nums.sort(
)for first in
range
(length-3)
:ifsum(nums[first:first+4]
)> target:
break
if first !=
0and nums[first]
== nums[first-1]
:continue
threesum = target-nums[first]
for second in
range
(first+
1, length-2)
:ifsum(nums[second:second+3]
)> threesum:
break
if second != first+
1and nums[second]
== nums[second-1]
:continue
twosum = threesum-nums[second]
thrid, four = second +
1, length -
1while thrid < four:
sum_two = nums[thrid]
+nums[four]
if sum_two == twosum:
[nums[first]
, nums[second]
, nums[thrid]
, nums[four]])
if sum_two >= twosum:
four -=
1while thrid+
1< four and nums[four]
== nums[four+1]
: four -=
1if sum_two <= twosum:
thrid +=
1while thrid+
1< four and nums[thrid]
== nums[thrid-1]
: thrid +=
1return res
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...