[LeetCode] 3Sum 以及unordered_set的学习

题目

---

思路

这道题是twosum的镜像问题，但是由于增加了一个数，难度增加了不少。有道类似的题曾经在大一上程序设计机考的时候见过，那时候是3重循环暴力求解，结果超时了。后来经过一点点优化，尽管时间复杂度仍是O(n^3)，但是还是通过了。当然这题难度更高，因为会出现重复的数字。

思路一：暴力求解

这个不必详述，直接三重循环。时间复杂度为O($n^3$)。

思路二：利用二分查找

第三重循环利用二分查找，当然，在开始循环之前就必须先把输入的序列进行排序，就可以将时间复杂度降到O($n^2logn$)

思路三：利用hash

由于第三重循环要查找的目标是可以通过前两层循环确定的目标确定下来的，所以就可以利用hash来免除最后一层循环，直接将时间复杂度降为O($n^2$)。而在C++ STL中的hash就是unordered_map、 unordered_set 、unordered_multimap 和 unordered_multiset 四个类。
关于具体使用我在后面详谈。
这个思路本来很好，但是因为这题要考虑到重复，于是用map或者set都会带来一些麻烦，虽然不是不能解决，但是实际上就感觉用了反而会带来许多麻烦，不是特别的优雅。比如说-2 1 1可以作为一个解，但是在hash中，因为不知道究竟有几个1，所以不知道这个可以不可以作为一个解。当然这个可以通过count来计算，但是这就增加了讨论的成本，使得代码不那么具有通用性，我希望能找到更好的解决办法。

思路四：两个指针，模仿数据库中的归并排序来进行并运算

这个来自于discussion中的思路，让我眼前一亮，这个思路和数据库中的归并排序中用的双指针同时遍历两个关系然后进行并运算。有重复的就跳过。这个思路很简单但是很好用，而且非常具有通用性，也不必要做任何无用的操作。具体思路就是在二层循环的时候同时从前后扫描（首先必须排序）。然后当找到解的时候，指针同时移动查找有没有重复的值（如果有重复的那必然在左右），这样子就能同时查重和找解，而且思路三中举例的问题在此处根本不需要考虑。以下是代码

class Solution {
public:
    vector<vector<int> > threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<vector<int> > v;
        int second, third;
        for (int first = 0; first < nums.size(); first++) {
            second = first +1;
            third = nums.size() -1;
            while (second < third) {
                if (nums[first] + nums[second] + nums[third] == 0) {
                    vector<int> temp;
                    temp.push_back(nums[first]);
                    temp.push_back(nums[second]);
                    temp.push_back(nums[third]);
                    v.push_back(temp);
                    // jump duplicate triplets
                    while (second < third && nums[second] == temp[1]) second++;
                    while (second < third && nums[third] == temp[2]) third--;
                } else if (nums[first] + nums[second] + nums[third] < 0) {
                    second++;
                    // if have duplicate secondary numbers, then jump it
                    while (second < third && nums[second] == nums[second -1]) second++;
                } else {
                    third--;
                    // if have duplicate tertiary numbers, then jump it                    
                    while (second < third && nums[third] == nums[third +1]) third--;
                }
            }
            while (first + 1 < nums.size() && nums[first] == nums[first +1]) first++;
        }
        return v;    
    }
};

unordered_set使用心得

虽然最后没有采用这个类，但是学习了一下这个类的用法。虽然是set，但是允许重复，采用的是拉链法解决冲突。但是在遍历的时候是不会重复输出的。值得一提的是，利用iterator遍历的时候，输出的结果是降序的，这和set底层的红黑树不太一样。