THREE SUM

3Sum问题

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:

[ [-1, 0, 1], [-1, -1, 2]]

解题思路1：

其实我自己的解决思路是暴力破解，先对数组进行排序，然后在三个while循环，定义三个循环变量i，j，k来求解，每求到一组解之后，将这个解保存到map < pair<int,int>,int > 中,三个数据分别是求得的三个解，如果已经存在，则不存储，继续寻找下一组解，这样的结果是算法复杂度O(n^3),在接受测试的时候，由于超时而失败。

解题思路2：

说实话，我觉得这种算法题，像我这样脑子比较笨的人，从来不能想到解决办法，于是我就网上百度一下，看到了别人说的2Sum问题，就是在一个数组中找到两个数使得他们的和是目标值t，这个方法就是先把数组从小到大排序，然后从数组的两端开始查找，假设为i和j，数组为a。2Sum的求解伪代码如下：

//2Sum的求解过程
//i是数组最左端，j是最右端，t是目标值
while(i!=j)
{
    if (a[i]+a[j]==t)
    {
        //保存这一组解
        rslt.store(a[i],a[j]);
        //同时i向右移动一次，继续求解
        ++i;
    }
    else if (a[i]+a[j]>t)
    {
        //两数之和大于目标值，需要缩小，于是把j向左移动
        --j;
    }
    else
    {
        //两数之和小于目标值，需要扩大，于是把i向右移动
        ++i;
    }
}
//返回结果
return rslt;

我从这个2Sum的解决方法上得到了启发，3Sum问题就是多了一个数据，于是我只需要在最外层加一个循环，从有序数组的最左端循环到最右端来求解，同时保证每次求到的解不要重复即可。因此，得到了3Sum的解决方法：

#include<algorithm>
vector<vector<int>> threeSum(vector<int>& nums) {
    //保存全部结果
    vector<vector<int>> rslt;
    //保存每一组解
    vector<int> tmp(3, 0);
    if (nums.size() < 3)
        return rslt;
    //快速排序成为有序数组
    sort(nums.begin(), nums.end());
    //最外层循环，从第一个数遍历到最后一个数
    auto it1 = nums.begin();
    while (it1 < nums.end()-1)
    {
        //it2和it3分别是2Sum问题中数组的两个端点
        auto it2 = it1 + 1;
        auto it3 = nums.end() - 1;
        //这个循环和2Sum问题循环基本一致
        while (it2 < it3)
        {
            if (*it1 + *it2 + *it3 > 0)
                it3--;
            else if (*it1 + *it2 + *it3 < 0)
                it2++;
            else
            {
                tmp[0] = *it1;
                tmp[1] = *it2;
                tmp[2] = *it3;
                rslt.push_back(tmp);
                //为了避免重复，如果it2之后的数和it2相同，则会跳过
                while (it2 < it3)
                {
                    if (*it2 == *(it2 + 1))
                        ++it2;
                    else
                        break;
                }
                ++it2;
            }
        }
        //同理，如果it1之后的数和it1所指的数相同，则跳过，避免重复
        while (it1 != nums.end() - 1)
        {
            if (*it1 == *(it1 + 1))
                ++it1;
            else
                break;
        }
        ++it1;
    }
    return rslt;
}

总结

我不知道这些人脑袋是如何思考来解决这些奇奇怪怪的算法题的，但是我知道我想破了脑袋都想不到这种解决办法，leetcode中的正则表达式匹配的那个问题我一直放着不做，因为我觉得我看别人的思路也看不懂，直接放弃了。我的结论是，我确实缺乏创新精神，尤其对于这种需要奇思妙想的算法题。因此，注定了我这种人只能在别人的的探照灯下前进。原谅我这么消极，喜欢了很久的妹子要离我而去了。