1. 程式人生 > >LeetCode18. 四數之和 python3

LeetCode18. 四數之和 python3

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