给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
https://leetcode-cn.com/problems/3sum/
思路:
1.首先需要考虑特殊情况(这不管是在解题还是解决工程问题都是需要考虑的)
对于数组长度n ,如果数组为null或者数组长度小于3,返回0.
2.对数组进行排序。
3.遍历排序后数组:
1)若nums[i]> 0:因为已经排序好,所以后面不可能有三个数加和等于0 ,直接返回结果。
2)对于重复元素:跳过, 避免出现重复解。
3)令左指针L=i+1,右指针R= n- 1,当L< R时,执行循环:
当nums[i] + nums[L] + nums[R]== 0 ,执行循环,判断左界和右界是否和下一位置重 复,去除重复解。并同时将L, R移到下一-位置,寻找新的解。
若和大于0,说明nums[R]太大, R左移。
若和小于0,说明nums[L]太小, L右移。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int a,b,c;
sort(nums.begin(), nums.end());
vector<vector<int>> result;
if (nums.size() < 3 || nums[0] > 0 || nums[nums.size()-1] < 0) return result;
for(int i=0; i<nums.size(); i++){
if(i>0 && nums[i]==nums[i-1])continue;//避免重复计算
int L = i+1;
int R = nums.size()-1;
while(L<R){
if(nums[i]+nums[L]+nums[R]==0){
result.push_back({nums[i],nums[L],nums[R]});
while(L<R && nums[L]==nums[L+1])
//考虑前面出现2,最后两个数为-1,则L和R交叉了,故加条件L<R
L++; //避免重复后移
while(L<R && nums[R]==nums[R-1])
R--;
if(L<R){ //防止下标越界
L++;R--; //sum=0后进行后续判断,指针均后移一位
}
}
if(L<R && nums[i]+nums[L]+nums[R]<0) //因为排好序的,和小于0就L右移
L++;
if(L<R && nums[i]+nums[L]+nums[R]>0) //L<R同样防止越界
R--;
}
}
return result;
}
};