leetcode解题报告15. 3Sum

题目地址

难度是medium

题目描述

输入一个整数数组，从中选择３个数，使得这三个数之和为０，这三个数称为一个三元组。现在要求所有满足要求的三元组。并且要求三元组之间不重复。（其中三元组里的元素是没有顺序的，即(1,2,3)和(3,2,1)是同一个三元组）

我的思路

首先暴力的办法是遍历所有三元组，找出满足要求的即可。但是一方面复杂度太高了O(n3)的复杂度。另一方面，如何进行三元组的去重呢？
对于第一个问题，可以考虑利用哈希表技巧，使得找一个数的复杂度从O(n)变成O(logn)，从而使得复杂度从O(n3)变成O(n2logn)
对于第二问题，想了好久，尝试了利用stl里的set来去重，但是不太会用，比较麻烦，放弃了。后来想到，三元组的比较麻烦在于顺序问题，可以先要求，三元组里的元素是升序的，那可以先确定第一个位置的元素，然后再确定第二个位置的元素，最后确定第三个位置的元素。如是得到的三元组是唯一的。具体操作是先对数组排序，然后第一个for循环确定第一个位置的元素，然后第二个for循环确定第二个位置的元素。要求是大于第一个位置元素。如是就可以判断第三个元素，要求在数组里，并大于等于第二个元素(如果等于，要保证原来数组里有多个该元素)，且三者和为零。

我的代码

#include <algorithm>#include <map>using namespace std;class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        vector<vector<int>> result;        //处理不满足要求的情况        if (nums.size() < 3) {            return result;        }        //要排序，避免改变输入，新建一个数组        vector<int> t_nums(nums.begin(), nums.end());        sort(t_nums.begin(), t_nums.end());        //建立哈希表，加速在一个数组里找一个数        map<int, int> my_map;        for (int i = 0; i < t_nums.size(); i++) {            if (my_map.count(t_nums[i]) > 0) {                my_map[t_nums[i]] += 1;            } else {                my_map[t_nums[i]] = 1;            }        }        int pre = t_nums[0]+1;//初始化成和第一个数不一样        for (int i = 0; i < t_nums.size()-1; i++) {      　　//避免重复            if (t_nums[i] == pre) {                continue;            }            pre = t_nums[i];            my_map[t_nums[i]] -= 1;            int pre_2 = t_nums[i+1] + 1;            for (int j = i+1; j < t_nums.size(); j++) {                if (t_nums[j] == pre_2) {                    continue;                }                pre_2 = t_nums[j];                my_map[t_nums[j]] -= 1;                int target = -(t_nums[i]+t_nums[j]);                if (target >= t_nums[j] && my_map.count(target) > 0 && my_map[target] > 0) {                    vector<int> temp;                    temp.push_back(t_nums[i]);                    temp.push_back(t_nums[j]);                    temp.push_back(target);                    result.push_back(temp);                }                my_map[t_nums[j]] += 1;            }            my_map[t_nums[i]] += 1;        }        return result;    }};

阅读官方题解

没有官方题解。网上有个一个O(n2)的思路，大体是先确定第一个位置元素。然后从数组的头尾开始分别找第二个位置元素和第三个位置元素。

思想核心总结

注意思考问题，不要过于依赖stl