Leetcode 15. 3Sum

https://leetcode.com/problems/3sum/

#include<algorithm>
#include<vector>
using namespace std;
class Solution {
public:
    void _sum(vector<int>& nums,int l,vector<vector<int>> &ans,int target,vector<int> &x,int n) {
        int L=nums.size();
        if(n>L-l || n*nums[l] > target || n*nums.back()<target) return;
        if(n==2){
            int r=nums.size()-1;
            while(l<r){
                if(nums[l]+nums[r]>target) --r;
                else if(nums[l]+nums[r]<target) ++l;
                else{
                    x.push_back(nums[l]);
                    x.push_back(nums[r]);
                    ans.push_back(x);
                    x.pop_back();
                    x.pop_back();
                    ++l;
                    --r;
                    while(l<r && nums[l]==nums[l-1]) ++l;
                    while(l<r && nums[r]==nums[r+1]) --r;
                }
            }
        }
        else{
            
            for(int i=l;i<=L-n;++i){
                if(i!=l && nums[i]==nums[i-1]) continue;
                x.push_back(nums[i]);
                vector<vector<int>> tmp;
                _sum(nums,i+1,tmp,target-nums[i],x,n-1);
                for(auto &xx:tmp)
                    ans.push_back(xx);
                x.pop_back();
            }
        }
        return;

    }
    vector<vector<int>> threeSum(vector<int>& nums) {
        /*
        当时2个数加起来为target时，可用逼近法【每一组的值都不可能有一个一样的，否则会重复】
        其他情况，不能重复使用与前一个位置的值一样的数来进行搜索【第l个一直可能】
        */

        sort(nums.begin(),nums.end());
        vector<vector<int>> ans;
        vector<int> x;
        _sum(nums,0,ans,0,x,3);
        return ans;
    }
};

python版本

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        ans=[]
        nums.sort()
        L=len(nums)
        for i in range(L):
            if i==0 or nums[i]!=nums[i-1]:
                l,r=i+1,L-1
                while  l<r:
                    x=nums[l]+nums[r]+nums[i]
                    if x==0:
                        ans.append([nums[i],nums[l],nums[r]])
                        l+=1
                        while l<r and nums[l]==nums[l-1]:
                            l+=1
                    elif x>0:
                        r-=1
                    else:
                        l+=1
        return ans