题目:
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请你返回所有和为 0
且不重复的三元组
思路:这道题目使用哈希法并不十分合适,因为在去重的操作中有很多细节需要注意,在面试中很难直接写出没有bug的代码。
使用双指针法比较高效。
首先将数组排序,用一层for循环,i从下标0的地方开始,同时下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上。依然还是在数组中找到 abc 使得a + b +c =0,我们这里相当于 a = nums[i],b = nums[left],c = nums[right]。
如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。
如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。
去重操作:对a去重,判断nums[i]与nums[i-1]是否相等,若相等则跳过本次循环。对b去重,判断nums[left]与nums[left+1]是否相等,若相等则跳过nums[left+1]。对c去重,判断nums[right]与nums[right-1]是否相等,若相等则跳过nums[right-1]。
代码:
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result=new ArrayList<>();
Arrays.sort(nums);
//a+b+c=0,先对a去重,再对b,c去重
for(int i=0;i<nums.length;i++){
if(nums[i]>0){
return result;
}
if(i>0&&nums[i]==nums[i-1]){//对a去重
continue;
}
int left=i+1;
int right=nums.length-1;
while(right>left){
int sum=nums[i]+nums[left]+nums[right];
if(sum>0){
right--;
}
else if(sum<0){
left++;
}
else{
result.add(Arrays.asList(nums[i],nums[left],nums[right]));
while(right>left&&nums[left]==nums[left+1])//对b去重
left++;
while(right>left&&nums[right]==nums[right-1])//对c去重
right--;
left++;
right--;
}
}
}
return result;
}