【每日一题】LeetCode. 15. 三数之和

每日一题，防止痴呆 = =

一、题目大意

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum

二、题目思路以及AC代码

先分享一个自己的最后没得出结果的思路吧… 可能这段时间一会做题一会不做题，脑子有点慢 = =。我一开始的想法是因为是三个数相加得0，那么其实确定两个数就是可以的，那么我们直接遍历两个数，然后第三个数利用哈希检查是否在数列里就可以了，这样也就是O(N^2)的复杂度。

但是题目要求结果中不能包含重复的三元组，这块卡了我好长时间，顺着这个思路，只能考虑怎么用哈希来检查重复三元组了，因为为了保证算法O(N^2)的复杂度，只能让这部分的检查复杂度为O(1)，想了一下， 其实检查重复三元组只需要检查其中两个数是否相同就行了，这样我考虑把两个数构造成字符串，再继续用哈希来检查，然而结果超时了 = =，可能涉及到字符串就比较慢了，我也没有考虑出别的编码方法，可以用来检查两个三元组中是否有两个数相等。

思路：

下面就说一下正确的思路。涉及到重复的情况，我们可以考虑排序，因为排完序后，重复的元素就都到一起了，那样我们就可以通过查看前后元素来去除重复了。我们首先对数组进行排序，然后枚举第一个元素，如果遇到当前元素和前一个元素重复，那么就跳过。

然后我们用双指针，令第三个元素的指针指向最后一个元素，然后我们枚举第二个元素，当三者相加大于0时，就令第三个元素的指针向左移动，直到和小于等于0，进行判断，如果等于0，那么把当前的三元组加入结果。可以双指针做的原因是，因为数组是排好序的，那么如果a + b + c > 0，那么随着a不变，b向后移动，a + b + c肯定不会等于0，所以当前的c是不可能的，就可以直接向左移动。上述的a, b, c指的是第一、第二、第三个元素。

AC代码

#define MAX_INT 2147483647

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;

        int n_size = nums.size();
        if (n_size < 3) return res;

        sort(nums.begin(), nums.end());

        for (int i=0;i<n_size;i++) {
            if (i != 0 && nums[i] == nums[i-1]) continue;
            int k = n_size - 1;
            for (int j=i+1;j<n_size;j++) {
                if (j == i+1 || nums[j] != nums[j-1]) {
                    while (k > j && nums[i] + nums[j] + nums[k] > 0) {
                        k--;
                    }
                    if (k <= j) break;
                    if (nums[i] + nums[j] + nums[k] == 0) {
                        vector<int> tmp;
                        tmp.push_back(nums[i]);
                        tmp.push_back(nums[j]);
                        tmp.push_back(nums[k]);
                        res.push_back(tmp);
                    }
                }
            }
        }

        return res;
    }
};

如果有问题，欢迎大家指正！！！