15. 3Sum(M)

题目描述

给定一个长度为n的数组，找出3个数a,b,c，使之满足a+b+c=0这个等式。找出满足这个条件的所有不重复集合。原题以及例子如下，

题目分析

PS：答案的集合不要求是有序的。

如果是暴力破解的话，那么这道题的时间复杂度为O(n^3)，穷举出所有的例子，但肯定是会超时的。我们可以做出这么一个转换，a+b=-c。做过Two Sum的朋友，肯定一下子就能想到如何以O(n)的时间复杂度来解决这个问题。那么这道题便迎刃而解了。
我在这里简单介绍下，如何以O(n)的时间复杂度解决a+b=-c这个问题。有两种方法，一是直接使用STL中的HashMap，这个我就不介绍了，相信大家看了讨论区的答案一下就知道了；二是使用两个指针的方法。在这里我们假定已知的数组已经是有序的(这里的时间复杂度为O(n^2)，所以可以在开始之间进行一个快排)。用两个指针，分别指向数组的头和尾，如果tail+head<-c，那么便把头指针想后移动；如果tail+head>-c，那么就把尾指针向前移动。显然，当我遍历完该数组后，便可得到所有的答案。此题解答源码如下，

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());

        vector<vector<int> > res;

        bool findOne = false;
        int l = nums.size();
        for (int i = l - 1; i >= 2; --i) {
            if (i != l - 1 && nums[i] == nums[i + 1]) {
                continue;
            }
            int goal = -nums[i];
            findOne = false;
            int first = 0;
            int last = i - 1;
            int preFirst = 0;
            while (first != last) {
                if (nums[first] + nums[last] < goal) {
                    ++first;
                } else if (nums[first] + nums[last] > goal) {
                    --last;
                } else {
                    if (findOne && (preFirst == nums[first])) {
                        ++first;
                        continue;
                    }
                    vector<int> temp = {nums[first], nums[last], nums[i]};
                    res.push_back(temp);
                    preFirst = nums[first];
                    ++first;
                    findOne = true;

                }
            }
        }
        return res;
    }
};