首页 > 代码库 > leetcode_18 4Sum
leetcode_18 4Sum
Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.
Note: The solution set must not contain duplicate quadruplets.
For example, given array S = [1, 0, -1, 0, -2, 2], and target = 0. A solution set is: [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]
找到数列中所有和等于目标数的四元组,需去重.多枚举一个数后,参照3Sum的做法,O(N^3)
python实现
class Solution(object): def fourSum(self, nums, target): nums.sort() res = [] length = len(nums) for i in range(0, length - 3): if i and nums[i] == nums[i - 1]: continue for j in range(i + 1, length - 2): if j != i + 1 and nums[j] == nums[j - 1]: continue sum = target - nums[i] - nums[j] left, right = j + 1, length - 1 while left < right: if nums[left] + nums[right] == sum: res.append([nums[i], nums[j], nums[left], nums[right]]) right -= 1 left += 1 while left < right and nums[left] == nums[left - 1]: left += 1 while left < right and nums[right] == nums[right + 1]: right -= 1 elif nums[left] + nums[right] > sum: right -= 1 else: left += 1 return res
leetcode_18 4Sum
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。