描述
给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。
数据范围:0 \le n \le 10000≤n≤1000,数组中各个元素值满足 |val | \le 100∣val∣≤100
空间复杂度:O(n^2)O(n2),时间复杂度 O(n^2)O(n2)
注意:
- 三元组(a、b、c)中的元素必须按非降序排列。(即a≤b≤c)
- 解集中不能包含重复的三元组。
例如,给定的数组 S = {-10 0 10 20 -10 -40},解集为(-10, -10, 20),(-10, 0, 10)
思路:类似滑动窗口,在固定一个数的情况下,缩小左右两个窗口边界
import java.util.*;
public class Solution {
public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
ArrayList<ArrayList<Integer>> list = new ArrayList<>();
Arrays.sort(num);
for(int k = 0;k<num.length;k++){
if(num[k]>0) break;
if(k>0&&num[k]==num[k-1]) continue;
int i = k+1;
int j = num.length-1;
while(i<j){
int sum = num[k]+num[j]+num[i];
if(sum>0){
while(i<j&&num[j]==num[--j]);//右边界缩小的同时排除重复值
}else if(sum<0){
while(i<j&&num[i]==num[++i]);
}else{
list.add(new ArrayList(Arrays.asList(num[k],num[i],num[j])));
while(i<j&&num[i]==num[++i]);
while(i<j&&num[j]==num[--j]);
}
}
}
return list;
}
}