leetcode解题报告15. 3Sum
难度是medium
题目描述
输入一个整数数组,从中选择3个数,使得这三个数之和为0,这三个数称为一个三元组。现在要求所有满足要求的三元组。并且要求三元组之间不重复。(其中三元组里的元素是没有顺序的,即(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