三数之和
题目描述
给你一个包含 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]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
1、要求不重复—> 去重操作 —>双指针法
2、三数之和–>两数之和 (-a)=b+c
先将nums数组排序
第一层for循环:i–>对nums数组的每个元素a进行循环,寻找b+c=(-a)的b和c
寻找b、c时—>使用双指针(left,right对应nums数组中的b和c)
b+c>(-a) right–;
b+c<(-a) left++;
代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
//先排序
Arrays.sort(nums);
List<List<Integer>> results = new ArrayList<List<Integer>>();
for(int a=0;a<nums.length;a++){
if(nums[a]>0) {return results;}
//if (nums[a] == nums[a+1]){} 漏掉-1,-1,2
if(a>0 && nums[a] == nums[a-1]){continue;} //continue 跳出这一次的for循环
int left=a+1;
int right=nums.length-1;
while(left<right){
if(nums[left]+nums[right]>(-nums[a])){
right--;
}else if(nums[left]+nums[right]<(-nums[a])){
left++;
}else {
List<Integer> res = new ArrayList<>();
res.add(nums[a]);
res.add(nums[left]);
res.add(nums[right]);
results.add(res);
//找到一个三元组,进行去重
while(right > left && nums[right] == nums[right-1]) {right--;} //必须写 right>left,溢出
while(right > left && nums[left] == nums[left+1]) {left++;}
//找到答案 逼近
right--;
left++;
}
}
}
return results;
}
}
注意
1、去重复逻辑如果放在找到三元组之前,0,0,0 的情况,可能直接导致 right<=left 了,从而漏掉了 0,0,0 这种三元组
2、continue 的使用:跳过本次循环体中余下尚未执行的语句,立即进行下一次的循环条件判定
3、注意while()中,当++,–时溢出情况判断
需要写left<right判断