LeetCode 15. 3Sum

求无序数列中三个数和为0的所有情况。

 

思路：排序。外层循环fix第一个数O(n)。内层循环O(n)同时寻找第二个和第三个数和为第一个数的相反数：考虑到不能出现重复情况，所以内层循环每次从第一个数的下一个值进行循环；两个指针分别指向第一个数的下一个数，和序列的最后一个数，如果三个数之和大于0，右指针向左，反之左指针向右。

注意：为避免出现重复，除序列中的首元素，如果当前fix的第一个数和序列中的前一个数相同，则跳过；

 

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        int n=nums.size(),l,r;
        vector<int> temp;
        vector<vector<int>> ans;
        for (int i=0;i<n-2&&nums[i]<=0;){
            l=i+1;r=n-1;
            while (l<r){
                while (nums[i]+nums[l]+nums[r]!=0 && l<r){
                    while (nums[i]+nums[l]+nums[r]>0 && l<r) r--;
                    while (nums[i]+nums[l]+nums[r]<0 && l<r) l++;
                }
                if (nums[i]+nums[l]+nums[r]==0 && l<r) {
                   temp.push_back(nums[i]);
                    temp.push_back(nums[l]);
                    temp.push_back(nums[r]);
                    ans.push_back(temp);    
                    temp.clear();
                    l++;
                    while(nums[l]==nums[l-1]&&l<n-1) l++;
                }//if
            }//while
            i++;
            while(nums[i]==nums[i-1]&&i<n-2) i++; 
        }//for
        return ans;
    }
};