15.三数之和
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
思路:该题与之前的两数之和有一定的相似之处,也是可以用多个指针来寻找三数之和为0的三元组。但是暴力枚举会导致解题超时,因此需要去除掉一些不可能的组合。比如将数组按升序排列,当nums[i]>0时,必然不可能有三数之和等于零的情况。因此可以依据这些限制条件来减少程序运行时间。
class Solution {
public List<List<Integer>> threeSum(int[] nums){
//创建二维列表
List<List<Integer>> ans=new ArrayList<>();
//如果数组长度小于3或者为空则直接返回
if(nums.length<3||nums == null){
return ans;
}
//对nums进行排序
Arrays.sort(nums);
// 遍历nums数组
for(int i=0;i<nums.length;i++){
//如果nums[i]>0后续不存三数之和等于0的情况,则直接退出循环
if(nums[i]>0){
break;
}
//如果有重复的nums[i]则直接跳过避免重复的数组
if(i>0&&nums[i]==nums[i-1]){
continue;
}
//定义另外两个指针
int j=i+1;
int k=nums.length-1;
//限制条件:左指针小于右指针
while(j<k){
//分为三种情况:和小于0则需要更大的值即移动左指针;
// 和大于0则需要更小的值即移动右指针;
//和等于0且左指针小于右指针则将三个数添加到列表中,将两个指针相向移动;
if(nums[j]+nums[i]+nums[k]<0){j++;}
if(nums[j]+nums[i]+nums[k]>0){k--;}
if(nums[j]+nums[i]+nums[k]==0&&j<k){
ans.add(Arrays.asList(nums[i],nums[j],nums[k]));
while (j<k && nums[j] == nums[j+1]) {j++;}// 去重
while (j<k && nums[k] == nums[k-1]) {k--;} // 去重
j++;
k--;
}
}
}
return ans;
}
}