leetcode刷题记录:（15） 三数之和

 难度：medium

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

---

个人思路，首先把数组从小到大排序，然后固定一个值使其从左往右遍历，该值右边第一个位置为left，表尾为right

如果三个值相加小于0，说明left的值不够大，向右移动left；相加大于0，说明right的值不够小，向右左移动right——直到left和right相遇

对于排除重复答案，直接用not in判断

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = []
        i = 0
        nums.sort(reverse = False)
        while i <= len(nums)-3:
            left = i+1
            right = len(nums) - 1
            while left < right:
                if nums[i]+nums[left]+nums[right] == 0:
                    if [nums[i],nums[left],nums[right]] not in res:
                        res.append([nums[i],nums[left],nums[right]])
                    left+=1
                    right-=1
                else:
                    if nums[left]+nums[right]+nums[i] < 0:
                        left+=1
                    else: right-=1
            i+=1
        return res

---

 根据解答区回答修改，思路大致相同，优化有：对于相同的固定值，直接使用continue跳过本次循环，LR也是通过相同思路改变

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = []
        i = 0
        nums.sort(reverse = False)
        while i <= len(nums)-3:
            if i > 0 and nums[i] == nums[i-1]:
                i+=1
                continue
            left = i+1
            right = len(nums) - 1
            while left < right:
                if nums[i]+nums[left]+nums[right] == 0:
                    res.append([nums[i],nums[left],nums[right]])
                    while left < right and nums[left]==nums[left+1]:
                        left+=1
                    while left < right and nums[right]==nums[right-1]:
                        right-=1
                    left+=1
                    right-=1
                elif nums[left]+nums[right]+nums[i] < 0:
                        left+=1
                else: right-=1
            i+=1
        return res