题目(难度:中等):
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解法:
暴力求解的算法复杂度是,稍微简单一点的思路是先将Nums数组从小到大排序利用三个变量k、i、j,其中k在数组最左端,i为k+1,j为数组右端,比较条件为(i< j),在此之前的条件是nums[k]不能大于0,因为数组是排好序的,一旦nums[k]>0,,意味着三个数相加一定>0,不符合条件。除此之外,还要去除重复的三元组,当nums[i]==nums[++i]或者nums[j]==nums[--j]时,需要跳过该元素。
代码实现:
package Middle;
import java.util.*;
/**
* Created by lizeyang on 2019/9/1.
*/
public class Solution5 {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> result = new ArrayList<>();
for(int k=0;k<nums.length-2;k++){
if(nums[k]>0){break;}
if(k > 0 && nums[k]==nums[k-1]){continue;}
int i = k+1,j = nums.length-1;
while(i<j){
int sum = nums[k]+nums[i]+nums[j];
if(sum<0){
while(i<j&&nums[i]==nums[++i]);
}else if(sum>0){
while(i<j&&nums[j]==nums[--j]);
}else{
result.add(new ArrayList<>(Arrays.asList(nums[k],nums[i],nums[j])));
while(i<j&&nums[i]==nums[++i]);
while(i<j&&nums[j]==nums[--j]);
}
}
}
return result;
}
}
测试用例:
算法分析:
时间复杂度,额外空间复杂度O(1)