leetcode 15. 三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

注意分析问题，i顺序遍历；j也顺序遍历和k逆序遍历，这样j和k相遇即可结束本次循环，可以大大降低复杂度；第一次做的时候选择i，j，k都是顺序遍历，理论上复杂度O(n3)，怪不得超时。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> v;
        vector<int> vec(3);
        if(nums.size()<3) return v;
        sort(nums.begin(),nums.end());//先转换成有序向量
        int sum=0,m=0;
        bool flag = 1;
        for(int i=0;i<nums.size()-2;++i){
            m=0-nums[i];
            int j=i+1,k=nums.size()-1;
            while(j<k){
                sum=nums[j]+nums[k];//转换成两数之和的问题
                if(sum==m){
                   vec[0]=nums[i];
                   vec[1]=nums[j++];
                   vec[2]=nums[k--];
                   v.push_back(vec);
                while(j<k&&nums[j]==nums[j-1]) j++;//j跳过重复元素
                while(j<k&&nums[k]==nums[k+1]) k--;//k跳过重复元素
                }
                else if(sum>m) k--;//sum(两数和)较大时，k左移
                else j++;//sum较小时，j右移
            }
            while(i<nums.size()-2&&nums[i]==nums[i+1]) i++;//i跳过重复元素
        }
        return v;
    }
};