题目:给你一个包含 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
10^5 <= nums[i] <= 10^5
题目要求找出和为0的三个数字,如果直接使用三个for循环,会导致时间复杂度很高,比如3000的三次方,时间复杂度有10^9,容易超时。在对数组进行排序的基础上,这时候可以考虑先固定一个数字,那么另外两个数字就是这个数字的负数,另外两个数字要比固定的这个数大,一个从左边开始,一共从右边开始,遍历整个数组,找出符合条件的数。
大概思路如上图所示,用for循环对固定的 那个遍历,用while循环另外两个指针。代码如下:
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList<>();
for (int i = 0; i < nums.length-2; i++) {
//重复的数字跳过
if(i>0 &&nums[i] == nums[i-1]){
continue;
}
//如果第一个数都大于0了,那后面的都大于0,肯定不符合条件
if(nums[i] > 0){
break;
}
int left = i+1;//左指针
int right = nums.length-1;//右指针
int target = -nums[i];
while(left<right){
int sum = nums[left] + nums[right];
if(sum == target){
res.add(Arrays.asList(nums[i],nums[left],nums[right]));
while(left<right && nums[left]==nums[left+1]){
left++;
}
while (left<right && nums[right] == nums[right-1]){
right--;
}
left++;
right--;
}else if(sum<target){
left++;
}else {
right--;
}
}
}
return res;
}
说明一下Arrays.asList() :
该方法是将数组转化成List集合的方法。
List<String> list = Arrays.asList("a","b","c");