LeetCode18. 四數之和 python3
阿新 • • 發佈:2018-12-12
給定一個包含 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] ]
class Solution: def fourSum(self, nums, target): """ :type nums: List[int] :type target: int :rtype: List[List[int]] """ nums.sort() num_save = [] if len(nums) < 4 or nums == []: return [] else: if len(nums) == 4 and sum(nums) == target: num_save.append(nums) return num_save else: for i in range(0, len(nums)-2): for j in range(i+1, len(nums)-1): l = j+1 r = len(nums) - 1 while l < r: sums = nums[i] + nums[j] + nums[l] + nums[r] if sums == target: num_save.append([nums[i], nums[j], nums[l], nums[r]]) break elif sums < target: l += 1 else: r -= 1 num_saves = list(set([tuple(t) for t in num_save])) return num_saves