NC54 数组中相加和为0的三元组

描述

给出一个有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)

注意：

1. 三元组（a、b、c）中的元素必须按非降序排列。（即a≤b≤c）
2. 解集中不能包含重复的三元组。

例如，给定的数组 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;
    }
}