LeetCode-15 三数之和
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
**注意:**答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
提示:
3 <= nums.length <= 3000
-105 <= nums[i] <= 105
solution
采用双指针的思想,
- 排序,并初始化向量res
- 设当前指向为i,其中j指向i的右边,k指向vector的最后一个位置,此时可以保证
nums[i]<= nums[j]<= nums[k]
- 当j<k时进入循环
- 当
nums[i]
大于0时,nums[i] + nums[j] + nums[k]>0
必然存在,退出循环 - 若出现
nums[i]=nums[i - 1]
,则可能导致res有重复元素,则执行continue操作,使同一个数字只计算一次 - 令
sum = nums[i] + nums[j] + nums[k]
,若sum<0
,则将j向右移动,若sum>0
,则将k向左移动,若sum=0
,则放入向量res中,并去除j右侧的重复元素和k左侧的重复元素 - 重复3-6步,直到j>=k时退出循环
#include <vector>using namespace std;//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public:vector<vector<int> > threeSum(vector<int> &nums) {int l = nums.size();vector<vector<int> > res;sort(nums.begin(), nums.end());for (int i = 0; i < l - 2; ++i) {if (nums[i] > 0) {break;}if (i - 1 >= 0 && nums[i] == nums[i - 1]) {continue;}int j = i + 1, k = l - 1;while (j < k) {int sum = nums[i] + nums[j] + nums[k];if (sum < 0) {while (j < k && nums[j] == nums[++j]) {//j++;}} else if (sum > 0) {while (j < k && nums[k] == nums[--k]) {//k--;}} else {res.push_back({nums[i], nums[j], nums[k]});while (j < k && nums[j] == nums[++j]) {//j++;}while (j < k && nums[k] == nums[--k]) {//k--;}}}}return res;}
};
//leetcode submit region end(Prohibit modification and deletion)int main() {Solution solution;vector<int> vec = {-1, 0, 1, 2, -1, -4};solution.threeSum(vec);
}