15. 三数之和
给你一个包含 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]
输出:[]
核心:排序+双指针,在过程中去重
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<Integer> list = new ArrayList<>();
List<List<Integer>> lists =new ArrayList<>();
for (int i = 0; i < nums.length-2; i++) {
//nums[i]>0提前退出,后面所有数都大于0无意义
if (nums[i]>0){
break;
}
//对i进行去重
if (i>0&&nums[i]==nums[i-1]){
continue;
}
//j为左指针,k为右指针
int j=i+1,k=nums.length-1;
//循环保证一个i多次判断直到j<k
while (j<k){
int sum=nums[i]+nums[j]+nums[k];
//每jk指针前进去重一步就需要进行一次sum判断
if (sum<0){
//j前进或对j去重
while (j<k&&nums[j]==nums[++j]);
}else if (sum>0){
//k前进或对k去重
while (j<k&&nums[k]==nums[--k]);
}else {
list.add(nums[i]);
list.add(nums[j]);
list.add(nums[k]);
lists.add(list);
list=new ArrayList<>();
//由于每个i循环中,j和k都只能使用一次,因此使用完一次进行前进且一定要去重
//而如果一开始就有sum==0或者过程中未进行去重,进入该语句,j和k就需要在此处去重
while (j<k&&nums[j]==nums[++j]);
while (j<k&&nums[k]==nums[--k]);
}
}
}
return lists;
}
}