给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:暴力解(N^3)会超时,先排序(nlogn)转化成a+b=-c双指针法
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
if(nums == null && nums.length ==0) return list;
sort(nums,0,nums.length-1);
for(int i =0;i<nums.length;i++){
if(nums[i]>0) break;
int start = i+1; int end = nums.length-1;
int num = nums[i] *-1;
if(i!=0 && nums[i]==nums[i-1]) continue;
while(start<end){
if(nums[start] +nums[end] >num || (end!=nums.length-1 && nums[end]==nums[end+1])) end--;
else if(nums[start] +nums[end] <num || (start!=i+1&&nums[start]==nums[start-1])) start++;
else {
List<Integer> list1 = new ArrayList<>();
list1.add(nums[i]);
list1.add(nums[start]);
list1.add(nums[end]);
list.add(list1);
start++;
end--;
}
}
}
return list;
}
private void sort(int[] arr,int start,int end ){
if(start >end) return ;
int i = start;
int j =end;
int temp =arr[start];
while(i<j){
while(i<j && arr[j]>temp) j--;
while(i<j && arr[i]<=temp ) i++;
swap(arr,i,j);
}
swap(arr,start,i);
sort(arr,start,i-1);
sort(arr,i+1,end);
}
private void swap(int arr[] ,int i ,int j){
int temp =arr[i];
arr[i] =arr[j];
arr[j] =temp;
}
}