18. 四数之和

题目描述请点击查看LeetCode 题目描述

Python3 代码解答如下:

class Solution:
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        """
        思路：与3数之和，同一个思路，只是多一层循环，先假定前两个数为第i个和第i+1个
        """
        nums.sort()
        print(nums)
        res =[]
        for i in range(0,len(nums)-3):
            if i == 0 or nums[i]>nums[i-1]:
                for j in range(i+1, len(nums)-2):
                    #print(nums[i], nums[j])
                    if j == i+1 or nums[j]>nums[j-1]:
                        l = j+1
                        r = len(nums)-1
                        while l < r:
                            s = nums[i] + nums[j] + nums[l] +nums[r]
                            if s == target:
                                li = [nums[i], nums[j], nums[l],nums[r]]
                                if li not in res:
                                    res.append(li)
                                l +=1
                                r -=1
                                while l < r and nums[l] == nums[l-1]:
                                    l += 1
                                while r > l and nums[r] == nums[r+1]:
                                    r -= 1
                            elif s>target:
                                r -=1
                            else :
                                l +=1
        return res