Letcode数组题:15.三数之和

三数之和

题目:给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,
使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。

示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[ [-1, 0, 1],[-1, -1, 2]]

解法:利用双指针法思路
1.当 nums[k] > 0 时直接break跳出:因为 nums[j] >= nums[i] >= nums[k] > 0,即 3 个数字都大于 0 ,在此固定指针 k 之后不可能再找到结果了。
在这里插入图片描述
2.先将数组nums排序,排序后图片如下图,然后分别设置三个指针k,i,j, k为长度为数组索引的循环,从0到(len(nums)-1),i则为k+1,j为(len(nums)-1)
在这里插入图片描述
3.k为0时,当s=nums[k] + nums[i] + nums[j] < 0时,i值加1,并判断此时的nums[i]是否和nums[i-1]相等,如果相等,则i再加1,直到不相等或i大于j为止。当s=nums[k] + nums[i] + nums[j] > 0时,j值减1,并判断此时的nums[j]是否和nums[j+1]相等,如果相等,则j再减1,直到不相等或i大于j为止。当s=nums[k] + nums[i] + nums[j] == 0时,返回值[nums[k],nums[i] ,nums[j]],并分别将i加1,j减1,并跳过所有重复的nums[i]和nums[j],防止记录到重复组合。
在这里插入图片描述
4.当i与j相等时,k值加1,然后按照上面方法继续计算,代码如下:

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        res = []
        for k in range(len(nums)-2):
            if nums[k] > 0:
                break
            if k > 0 and nums[k] == nums[k-1]:
                continue
            i = k+1
            j = len(nums)-1
            while(i<j):
                s = nums[k] + nums[i] + nums[j]
                if s > 0:
                    j -= 1
                    while i < j and nums[j] == nums[j+1]:
                        j -= 1
                elif s < 0:
                    i += 1
                    while i < j and nums[i] == nums[i-1]:
                        i += 1
                else:
                    res.append([nums[k],nums[i],nums[j]])
                    i += 1
                    j -= 1
                    while i < j and nums[j] == nums[j + 1]:
                        j -= 1
                    while i < j and nums[i] == nums[i - 1]:
                        i += 1
        return res
发布了7 篇原创文章 · 获赞 0 · 访问量 630

猜你喜欢

转载自blog.csdn.net/qq_36506038/article/details/105320484
今日推荐