三数之和

/*
 * 思路：
 * 1. 使用Java Stream来简化代码流程。
 * 2. 首先对数组进行排序。
 * 3. 使用三指针方法，固定一个数，然后在剩余的数组中使用双指针来查找另外两个数。
 * 4. 遍历过程中，跳过重复的数以避免重复结果。
 */
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class ThreeSumStream {
    public List<List<Integer>> threeSum(int[] nums) {
        if (nums == null || nums.length < 3) return new ArrayList<>();
        Arrays.sort(nums); // 对数组进行排序
        return Arrays.stream(nums).boxed().flatMap(i -> {
            if (i > 0 && nums[i] == nums[i - 1]) return null; // 跳过重复的数
            int left = i + 1, right = nums.length - 1;
            List<List<Integer>> temp = new ArrayList<>();
            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];
                if (sum == 0) {
                    temp.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    while (left < right && nums[left] == nums[left + 1]) left++; // 跳过重复的数
                    while (left < right && nums[right] == nums[right - 1]) right--; // 跳过重复的数
                    left++;
                    right--;
                } else if (sum < 0) {
                    left++;
                } else {
                    right--;
                }
            }
            return temp.stream();
        }).distinct().collect(Collectors.toList());
    }

    public static void main(String[] args) {
        ThreeSumStream solution = new ThreeSumStream();
        int[] nums = {-1, 0, 1, 2, -1, -4};
        System.out.println(solution.threeSum(nums)); // 输出: [[-1, -1, 2], [-1, 0, 1]]
    }
}