leetcode解题报告15. 3Sum

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