算法设计与分析—三数之和

题目描述:

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

示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]

示例 2:
输入:nums = []
输出:[]

示例 3:
输入:nums = [0]
输出:[]

来源:力扣(LeetCode)


算法思路:

方法一: 暴力解法,三重循环,时间复杂度O(n^3) 这里没有去实现

方法二: 排序+双指针 时间复杂度为O(n^2)

1、先对list 数组进行排序,然后开始两层循环
2、第一层循环 i 从0 到 n-3,每次取i后 后面的数据开始循环,采用双指针left=i+1和right=n-1 ,直到left>right 循环结束
3、第一层忽略相邻重复元素的方法 是判断nums[i] 与 nums[i-1]是否相等,相等则跳过。第二层循环忽略相邻重复元素的方法是判断nums[left]==nums[left+1] 和 nums[right]==nums[right-1] 是否相等,相等则跳过。

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums=sorted(nums)
        list=[]
        length=len(nums)
        if length<3:
            return list
        for i in range(length-2):
            if nums[i]>0:
                return list
            if i>0  and nums[i]==nums[i-1]:
                continue
            left,right=i+1,length-1
            while left<right:
                if nums[i]+nums[left]+nums[right]==0:
                    list.append ([nums[i],nums[left],nums[right]])
                    while left<right and nums[left]==nums[left+1]:
                        left=left+1
                    while left<right and nums[right]==nums[right-1]:
                        right=right-1
                    left=left+1
                    right=right-1
                elif nums[i]+nums[left]+nums[right]<0:
                    left=left+1
                else:
                    right=right-1
        return list

猜你喜欢

转载自blog.csdn.net/qq_39740279/article/details/120455945