四数之和python(leetcode18)

#18. 四数之和

先来做15. 三数之和

例子：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

是15. 三数之和的升级版

class Solution(object):
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        n = len(nums)
        ans = []
        nums = sorted(nums)
        if (n < 4):
            return []
        for i in range(0, n):
            if(i > 0 and nums[i] == nums[i - 1]):
                continue
            # j的循环是与三数之和的差别
            for j in range(i + 1, n):
                if(j > i + 1 and nums[j] == nums[j - 1]):
                    continue
                l, r = j + 1, n - 1
                while(l < r):
                    temp = nums[i] + nums[j] + nums[l] + nums[r]
                    if temp < target:
                        l += 1
                    elif temp > target:
                        r -= 1
                    else:
                        ans.append([nums[i], nums[j], nums[l], nums[r]])
                        while(l < r and nums[l] == nums[l + 1]):
                            l += 1
                        while(l < r and nums[r] == nums[r - 1]):
                            r -= 1
                        l += 1
                        r -= 1
        return ans