【LeetCode】15. 3Sum 三数之和

题目

15. 3Sum

思路

和2Sum有点类似，首先先排序，使其成为有序数组，之后选定当前为三数之中的一个，然后在其后的数字中用两个指针选数，注意，由于是数组是有序的，所以对于一个数来说，是O（n）的，其中需要注意的是不能有重复的组合，但是如果有重复的数，数可以重复用。由于是三个数，在同样的i只用一次，同一个i下，限制第二个数不能一样，就可以避免重复组合。
注意：不要在排序后直接处理重复数字，因为组合时还可能需要重复数字。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> vv;
        int n=nums.size(),tot=0;
        sort(nums.begin(),nums.end());
        for(int i=0;i<n;i++){
            if(i>0&&nums[i]==nums[i-1])continue;
            int j=i+1,k=n-1;
            while(j<k){
                if(j!=i+1&&nums[j]==nums[j-1]){
                    j++;
                    continue;
                }
                if(nums[j]+nums[k]==-nums[i]){
                    vector<int> v;
                    v.push_back(nums[i]);
                    v.push_back(nums[j]);
                    v.push_back(nums[k]);
                    vv.push_back(v);
                    j++;k--;
                }
                else if(nums[j]+nums[k]<-nums[i])j++;
                else k--;
            }
        }
        return vv;
    }
};