美文网首页
15. 三数之和

15. 三数之和

作者: 在做算法的巨巨 | 来源:发表于2018-09-06 18:48 被阅读0次

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。

注意:答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]


这道题的前提是需要将数组变得有序,这样才能进行后序的工作。
思路:
nums[i]+nums[k] > target ---> k--
nums[i]+nums[k] == target ---> k--,i++
nums[i]+nums[k] < target ---> i++

nums.sort()                
i=0
group = []
while (i <=len(nums)-2):
    j=i+1
    k=len(nums)-1
    target = 0-nums[i]
    while (j < k):
        if nums[j]+nums[k] == target:
            group.append([nums[i],nums[j],nums[k]])
            j=j+1
            k=k-1
            while (nums[j] == nums[j-1] and j<k):
                j=j+1    
        elif nums[j]+nums[k] > target:
            k=k-1
            while (nums[k] == nums[k+1] and j<k):
                k=k-1
        elif nums[j]+nums[k] < target:
            j=j+1
            while (nums[j] == nums[j-1] and j<k):
                j=j+1
    i=i+1
    while (nums[i] == nums[i-1] and i < len(nums)-2):
        i=i+1

相关文章

网友评论

      本文标题:15. 三数之和

      本文链接:https://www.haomeiwen.com/subject/dldxgftx.html