3Sum - Leetcode-CSDN博客

本文链接：https://blog.csdn.net/u013115185/article/details/49427121

题意：找出数组中三个数字之和为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;
    }
};