首页 > 学院 > 开发设计 > 正文

[LeetCode] 3Sum 解题报告

2019-11-08 01:08:04
字体:
来源:转载
供稿:网友

[题目] Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0 ? Find all unique triplets in the array which gives the sum of zero. Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is: [  [-1, 0, 1],  [-1, -1, 2] ]

[中文翻译] 给定n个整数的数组S,在S中是否存在三个元素a,b,c,使得a + b + c = 0 ?找到数组中所有和为零的不同的三元组。 注意:解决方案集不能包含重复的三元组。

例如, 给定数组S = [-1, 0, 1, 2, -1, -4],

解集为: [  [-1, 0, 1],  [-1, -1, 2] ]

[解题思路] 朴素的算法,枚举a, b, c。O(n3)的效率,加上处理重复的组合,可能不止O(n3)。

考虑一个更简单的问题,2Sum,且数组是有序的(假设从小到大)。 朴素的算法枚举a, b,效率O(n2)。但其实,有很多情况我们是不需要枚举的。 例如数组A, B, C, D, E, F, G, H。如果A+H>0,那么B~G+H一定>0(因为数组是有序的)。如果A+H<0,那么A+B~G一定<0。如果A+H=0,那么A+B~G以及B~G+H要么与A, H重复,要么不等于零。因此,只需要从两边开始向中间遍历即可,根据A+H与0比较的情况,判断哪边需要往中间缩。这种思想和[LeetCode] Container With Most Water 解题报告 中的思想很像,都是根据一些性质排除那些不必要的枚举。该方法的效率为O(n)。

那么回到我们的3Sum问题,其实只需要先确定一个数c,然后问题就转换为了2Sum问题,不同的是原来的要求是a+b=0,因此A+H是与0去比较,现在是要求a+b=-c,因此A+H是与-c去比较。

代码实现地比较复杂,主要是在处理重复三元组的问题。因为最后结果集可能是n3的数量级,不能在获得结果集之后剔除重复三元组,这样的效率至少是O(n3logn)的。因此,需要在枚举的时候就排除枚举到重复的三元组。

[C++代码]

class Solution {public: vector<vector<int>> threeSum(vector<int>& nums) { vector<vector<int>> res; sort(nums.begin(), nums.end()); for (int i = 1; i < nums.size(); i++) { int j = 0; if (i > 1 && nums.at(i - 1) == nums.at(i)) { if (nums.at(i - 2) == nums.at(i - 1)) continue; else { j = i - 1; } } int k = nums.size() - 1; for ( ; j < i; j++) { if (j > 0 && nums.at(j - 1) == nums.at(j)) continue; while (k > i && (nums.at(i) + nums.at(j) + nums.at(k) > 0)) k--; if (k > i && (0 == nums.at(i) + nums.at(j) + nums.at(k))) { vector<int> tmp; vector<int> last; if (res.size() > 0) { last = res.at(res.size() - 1); if (last.at(0) != nums.at(j) || last.at(1) != nums.at(i) || last.at(2) != nums.at(k)) { tmp.push_back(nums.at(j)); tmp.push_back(nums.at(i)); tmp.push_back(nums.at(k)); res.push_back(tmp); } } else { tmp.push_back(nums.at(j)); tmp.push_back(nums.at(i)); tmp.push_back(nums.at(k)); res.push_back(tmp); } int tmp2 = nums.at(k); while (k > i && (nums.at(k) == tmp2)) k--; } } } return res; }};
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表