题解 | #牛牛的三元组问题#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return int整型二维数组
     */
    public int[][] findTriplets (int[] nums) {
        // write code here
        int n = nums.length;
        Arrays.sort(nums);
        List<int[]> list = new ArrayList<>();
        for(int i = 0; i < n - 2; i++){
            //给三元组第一位去重
            while(i > 0 && i < n - 2 && nums[i] == nums[i - 1]) i++;
            int l = i + 1, r = n - 1;
            while(l < r){
                int sum = nums[i] + nums[l] + nums[r];
                if(sum < 0) l++;
                else if(sum > 0) r--;
                else{
                    list.add(new int[]{nums[i], nums[l], nums[r]});
                    l++;
                    //给三元组第二位去重
                    while(l < r && nums[l] == nums[l - 1]) l++;
                }    
            } 
        }
        int[][] ans = new int[list.size()][3];
        for(int i = 0; i < list.size(); i++){
            ans[i] = list.get(i);
        }
        return ans;
    }
}