描述
给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。
注意:
三元组(a、b、c)中的元素必须按非降序排列。(即a≤b≤c)
解集中不能包含重复的三元组。
例如,给定的数组 S = {-10 0 10 20 -10 -40},解集为(-10, -10, 20),(-10, 0, 10)
0 <= S.length <= 1000
示例1
输入:
[0]
返回值:
[]
示例2
输入:
[-2,0,1,1,2]
返回值:
[[-2,0,2],[-2,1,1]]
示例3
输入:
[-10,0,10,20,-10,-40]
返回值:
[[-10,-10,20],[-10,0,10]]
方法:
思路:
(1)首先对数组进行排序(从小到大)
(2)依次取出第 i 个数(i从0开始),并且不重复的选取(跳过重复的数)
(3)这样问题就转换为 2 个数求和的问题(可以用双指针解决方法)
==》数求和问题
(4)定义两个指针:左指针(left) 和 右指针(right)
(5)找出固定 left, 此时left所指的位置为数组中最小数,再找到两个数和 不大于 target 的最大 right 的位置
(6)调整 left 的位置(后移),求解和是否为 target O(n)
==》时间复杂度:O(nlogn) + O(n)
import java.util.ArrayList;
import java.util.Arrays;
public class Solution {
public ArrayList<ArrayList> threeSum(int[] num) {
ArrayList<ArrayList> list=new ArrayList<>();
if(num==null||num.length<3) return list;
Arrays.sort(num);//排序
for(int i=0;i<num.length-2;i++){
if(num[i]>0){//因为已经排好序,如果第i个大于0,则后面的都大于0
break;
}
if(i>0&&num[i]num[i-1]){
continue;//去重
}
int l=i+1;
int r=num.length-1;
while(l<r){
int sum=num[i]+num[l]+num[r];
if(sum0){
ArrayList li=new ArrayList<>();
li.add(num[i]);
li.add(num[l]);
li.add(num[r]);
list.add(li);
while (l<r&&num[l]==num[l+1]){
l++;
}
while (l<r&&num[r]==num[r-1]){
r–;
}
l++;
r–;
}else if(sum>0){
r–;
}else if(sum<0){
l++;
}
}
}
return list;
}
}