题目:
给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
java 代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
//先排好序
for(int i=0;i<nums.length;i++){
for(int j=i+1;j<nums.length;j++){
if(nums[i]>nums[j]){
int temp=nums[i];
nums[i]=nums[j];
nums[j]=temp;
}
}
}
List<List<Integer>> list=new ArrayList<>();
int i=0;
//先固定第一个数,然后根据三个数的和大于还是小于0移动后两个数
//下面出现的所有continue都是为了去重
while(i<nums.length){
if(nums[i]>0)
return list;
if(i>0&&nums[i-1]==nums[i]){
i++;
continue;
}
//第二个数在头部,第三个数在尾部
for(int j=i+1,k=nums.length-1;j<k;){
if(j>i+1&&nums[j-1]==nums[j]){
j++;
continue;
}
if(k<nums.length-1&&nums[k]==nums[k+1]){
k--;
continue;
}
int temp=nums[i]+nums[j]+nums[k];
if(temp>0)
k--;
else if(temp<0)
j++;
else{
List<Integer> list1=new ArrayList<>();
list1.add(nums[i]);
list1.add(nums[j]);
list1.add(nums[k]);
list.add(list1);
j++;
k--;
}
}
i++;
}
return list;
}
}
注意去除重复的结果。设一个满足条件的三元组<a,b,c>,如果有重复的三元组与之相同,则说明a,b,c中至少有一个元素的值在数组中出现至少两次。假如a的值2,在数组中出现多次,则其必然是连续的(数组已经排序),因此可以使用如上的方法去除重复的三元组。该方法时间复杂度O(nlogn)+O(n^2),空间复杂度为O(1)。