3Sum - Leetcode

题意：找出数组中三个数字之和为0的数对

For example, given array S = {-1 0 1 2 -1 -4}
A solution set is:
(-1, 0, 1)
(-1, -1, 2)

Leetcode第一题是Two Sum，经网上查看资料，kSum是一系列的算法题目。主要有三种解法：
第一种方法，暴力，枚举所有的k个数字之和，时间复杂度为O(n^k);
第二种方法，排序，首先将数组的元素从小到大排序，然后设置首尾指针遍历求和；对于2Sum的时间负责度是O(nlogn)；k>2时，时间复杂度是O(n^(k-1))；
第三种方法，Hash解法，在Two Sum这道题中用hashmap存储数组，每存储一个数字，就查找(target-i)数字在不在hashmap中，时间复杂度是O(n)。对于k>2时，时间复杂度是O(n^(k-1))；
这道题采用排序方法做，和2Sum的做法差不多，因为是三个数字，所以我们可以先固定一个数字，然后设置首尾指针遍历求和。

class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        vector<int> num;        //关联容器用于后面的去重        set<vector<int> > result;        vector<vector<int> > res;        //数组长度小于3时，返回空的res        if(nums.size()<3) return res;        num = nums;        //将数组从小到大排序        sort(num.begin(), num.end());        //如果数组中有重复的数字，经过排序之后重复的数字变为相邻。iCurrentValue变量的        //设置用于判断相邻的两个数字是否相等，如果相等则跳过循环中的操作，因为过程和前一        //个数字的过程是一样的，并且结果也是一样的。这个步骤为缩短运行时间有关键作用        int iCurrentValue = num[0];        //固定数字的范围从0到num.size()-2，因为后面要放首尾两个指针        for (int i = 0; i < num.size()-2; i++)        {            if (i&&iCurrentValue == num[i])                continue;            //首指针从固定数字的下一位开始            for (int p = i + 1, q = num.size() - 1; p < q;)            {                int n = num[i] + num[p] + num[q];                if (n == 0)                {                    vector<int> k{ num[i],num[p],num[q] };                    sort(k.begin(), k.end());                    result.insert(k);                    //在获得一个和为0的数对之后，pq值同时改变。因为此时为0，如果只移                    //动其中一个的位置的话肯定不会有0出现。                    p++;                    q--;                }                else if (n < 0) p++;                else q--;            }            iCurrentValue = num[i];        }        res=vector<vector<int> >(result.begin(), result.end());        return res;    }};

总结：这道题笔者刚开始做的时候，时间超出限制。虽然笔者思路与上面代码一样，但是过程不够精简，比如i,p,q起始位置以及范围，多出很多判断的地方，浪费了时间；再比如对于数组中相同的数字没有跳过，重复了整个过程浪费了时间。