leetcode-华为专题-15. 三数之和

 

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {

        vector<vector<int>> res;
        sort(nums.begin(), nums.end());  // 排序，方便后续处理

        for(int i = 0; i < nums.size(); i++){
            if(nums[i]>0)  // 第一个元素大于0，三数之和肯定大于0
                return res;
             // 去除重复，因为i-1肯定已经判断过了，如果nums[i]还与其相等，直接跳过
            if((i-1)>=0&&nums[i]==nums[i-1])
                continue; 
            int m = i+1;
            int n = nums.size()-1;
            while(m<n){ // 如果m==n，肯定不够两个数了
                if(nums[m]+nums[n]>(-nums[i]))
                    n--;
                else if(nums[m]+nums[n]<(-nums[i]))
                    m++;
                else{
                    
                    while(m<n&&nums[m]==nums[m+1]) // 去除重复
                        m++;
                    while(m<n&&nums[n]==nums[n-1])  // 去除重复
                        n--;
                    res.push_back({nums[i],nums[m],nums[n]});
                    m++;
                    n--;
                }
            }

        }
        return res;
    }
};