LeetCode第15题--三数之和

C++

使用set集合（有序去重）再用assign函数（vector的函数，将原来的vector清空，赋上新的vector）转回vector，这样做超出时间限制。可能跟set或者assign的底层实现有关在这里插入代码片
双指针思想：循环遍历i位置的元素，固定i元素，然后两指针指向左右两端，不断往里缩进。i与i+1元素相同直接跳过i+1因为解必重复，左右两指针与相邻重复，就额外的++或–
关于去重，要考虑到三个位置，循环的i，左指针l，右指针r都有可能与相邻的重复。因为题目要求的是不能重复解，因为排序之后，小元素在前，所以当解的第一个元素相同的时候（即i与i+1位置元素相同的时候），后面的左右指针指向的元素肯定与之前的相同

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size()<3) return ans;
        sort(nums.begin(),nums.end());
        int n = nums.size();
        for(int i = 0;i<n;i++){
            if(nums[i]>0) return ans;
            if(i>0 && nums[i]==nums[i-1]) continue;
            int l = i + 1, r = n - 1;
            while(l<r){
                int t = nums[i]+nums[l]+nums[r];
                if(t==0){
                    ans.push_back({nums[i], nums[l], nums[r]});
                    while(r>l && nums[r]==nums[r-1]) r--;	    //有重复的元素就再往里走
                    while(r>l && nums[l]==nums[l+1]) l++;
                    r--;
                    l++;				//找到一个答案之后就左右指针往里走
//这里必须放在判断重复的后面，如果先往里走了，可能与中间还没有判断的重复，有漏解
                                    }
                else if(t>0)
                r--;
                else
                l++;
            }
        }
        return ans;
    }
};

Python

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        ans = []
        if len(nums)<3: return ans
        nums.sort()
        n = len(nums)
        for i in range(n):
            l, r = i + 1, n - 1
            if nums[i] > 0: return ans
            if i>0 and nums[i] == nums[i-1]: continue #当这个位置重复时，也要判断重复
            while l<r:
                t = nums[i]+nums[l]+nums[r]
                if t==0:
                    ans.append([nums[i],nums[l],nums[r]])
                    while l<r and nums[l]==nums[l+1]: l+=1
                    while l<r and nums[r]==nums[r-1]: r-=1
                    l+=1
                    r-=1
                elif t<0: l+=1
                else: r-=1
        return ans