码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode 18-> 4Sum

时间:2019-03-13 19:57:01      阅读:159      评论:0      收藏:0      [点我收藏+]

标签:target   pre   4sum   span   color   object   solution   res   bsp   

 

class Solution(object):
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        
        l = len(nums)
        nums.sort()
        if l < 4:
            return []
        res = []
        for i in range(l-3):
            if i > 0 and nums[i] == nums[i-1]:
                continue
            elif nums[i] + nums[i+1] + nums[i+2] + nums[i+3] > target:
                break
            elif nums[i] + nums[l-3] + nums[l-2] + nums[l-1] < target:
                continue
            for j in range(i+1, l-2):
                if j > i+1 and nums[j] == nums[j-1]:
                    continue
                elif nums[i] + nums[j] + nums[j+1] + nums[j+2] > target:
                    break
                elif nums[i] + nums[j] + nums[l-2] + nums[l-1] < target:
                    continue
                k = j+1
                p = l-1
                s = target - nums[i] - nums[j]
                while k < p:
                    if nums[k] + nums[p] > s:
                        p -= 1
                    elif nums[k] + nums[p] < s:
                        k += 1
                    else:
                        res.append([nums[i], nums[j], nums[k], nums[p]])
                        while k < p and nums[k] == nums[k+1]:
                            k += 1
                        while k < p and nums[p] == nums[p-1]:
                            p -= 1
                        k += 1
                        p -= 1
        return res

 

leetcode 18-> 4Sum

标签:target   pre   4sum   span   color   object   solution   res   bsp   

原文地址:https://www.cnblogs.com/sea-stream/p/10525045.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!