leetcode 15. 3Sum 双指针

题目链接

给n个数， 找出三个数相加结果为0的所有的组， 不可重复。

用双指针的思想，O(n^2)暴力的找， 注意判重复。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int sz = nums.size();
        vector <vector<int> > ans;
        vector <int> tmp;
        sort(nums.begin(), nums.end());
        for(int i = 0; i<sz; i++) {
            if(i>0 && nums[i] == nums[i-1])
                continue;
            int l = i+1, r = sz-1;
            while(l<r) {
                int sum = nums[i]+nums[l]+nums[r];
                if(sum < 0) {
                    l++;
                } else if (sum>0) {
                    r--;
                } else {
                    tmp.push_back(nums[i]);
                    tmp.push_back(nums[l++]);
                    tmp.push_back(nums[r--]);
                    sort(tmp.begin(), tmp.end());
                    ans.push_back(tmp);
                    tmp.clear();
                    while(l<r&&nums[l-1]==nums[l])
                        l++;
                    while(r>l&&nums[r] == nums[r+1])
                        r--;
                }
            }
        }
        return ans;
    }
};