leetcode15——3Sum

题目大意：在给出的数组中找出所有三数相加和为0的数

分析：K sum问题，排序+双指针。为降低复杂度，固定答案的第一个数（第一层），从剩下的数（第二层）中双指针查找。要注意答案不能重复。

代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size() < 3)
            return ans;
        sort(nums.begin(),nums.end());
        for(int i = 0;i < nums.size() - 2 && nums[i] <= 0;i++){
            if(i > 0 && nums[i] == nums[i - 1])
                continue;
            int m = i + 1;
            int n = nums.size() - 1;
            while(m < n){
                if(nums[m] + nums[n] + nums[i] == 0){
                    vector<int> tmp;
                    tmp.push_back(nums[i]);
                    tmp.push_back(nums[m]);
                    tmp.push_back(nums[n]);
                    ans.push_back(tmp);
                    while(m < n && nums[m] == nums[m + 1]) //跳过重复数字
                        m++;
                    m++;
                    while(m < n && nums[n] == nums[n - 1])
                        n--;
                    n--;
                }
                else if(nums[m] + nums[n] + nums[i] < 0){
                    m++;   
                }
                else n--;
            }
        }
        return ans;
    }
};