题目描述:标签:nums,++,sum,--,while,result,周一 From: https://www.cnblogs.com/zeyangshuaige/p/17402785.html
给你一个整数数组 nums,请编写一个函数来判断其中是否存在三个元素,满足它们相加之和等于 0 。请返回所有满足条件且不重复的三元组。注意:答案中不可以包含重复的三元组。
设计思路:
首先对数组进行排序,然后使用双指针法进行遍历数组,将三数之和问题转化为两数之和问题,固定一个数作为基准数nums[i],然后在其后面的数中使用双指针法求出另外两个数字的和等于基准数的相反数即可。
程序流程图:
开始
定义二维数组result来存储所有符合条件的三元组
将数组nums进行排序
for i = 0 to n-2 do
if (i > 0 and nums[i] == nums[i-1]) continue
l = i + 1
r = n - 1
while l < r do
sum = nums[i] + nums[l] + nums[r]
if sum == 0 then
将当前符合条件的三元组加入result中
while l < r and nums[l] == nums[l+1] do l++
while l < r and nums[r] == nums[r-1] do r--
l++
r--
else if sum < 0 then l++
else r--
end for
返回result
结束
代码实现:
#include<iostream>
using namespace std;
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
sort(nums.begin(), nums.end());
int n = nums.size();
for (int i = 0; i < n-2; i++) {
if (i > 0 && nums[i] == nums[i-1]) continue;
int l = i + 1, r = n - 1;
while (l < r) {
int sum = nums[i] + nums[l] + nums[r];
if (sum == 0) {
result.push_back({nums[i], nums[l], nums[r]});
while (l < r && nums[l] == nums[l+1]) l++;
while (l < r && nums[r] == nums[r-1]) r--;
l++;
r--;
} else if (sum < 0) {
l++;
} else {
r--;
}
}
}
return result;
}
};