LeetCode15 - 三数之和

LeetCode15 - 三数之和

链接：https://leetcode-cn.com/problems/3sum
给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
注意：答案中不可以包含重复的三元组。 

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
示例 2：

输入：nums = []
输出：[]
示例 3：

输入：nums = [0]
输出：[]

解法：时间复杂度O(n^2)，空间复杂度O(1)，先排序，再双指针

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        if(nums == null || nums.length < 3) return result;

        // 先排序
        fastSort(nums, 0, nums.length-1);
        for(int i=0; i<nums.length-2; i++) {
            if(nums[i] > 0 || nums[nums.length-1] <0) {
                break;
            }
            if(i>0 && nums[i] == nums[i-1]) continue;
            int p=i+1, q=nums.length-1;
            while(p<q) {
                int sum = nums[i] + nums[p] + nums[q];
                if(sum == 0) {
                    List<Integer> temp = new ArrayList<>();
                    temp.add(nums[i]);
                    temp.add(nums[p]);
                    temp.add(nums[q]);
                    result.add(temp);

                    while(p<q && nums[p] == nums[p+1]) p++;
                    while(p<q && nums[q] == nums[q-1]) q--;
                    p++;
                    q--;
                }
                else if(sum<0) {
                    p++;
                }
                else {
                    q--;
                }
            }
        }
        return result;
    }

    private void fastSort(int[] nums, int start, int end) {
        int hi = end; 
        int lo = start;
        int key = nums[start];

        while(lo < hi) {
            while(lo < hi && key <= nums[hi]) {
                hi--;
            }
            swap(nums, lo, hi);
            while(lo < hi && nums[lo] <= key) {
                lo++;
            }
            swap(nums, lo, hi);
        }
        if(start < lo) fastSort(nums, start, lo-1);
        if(hi < end) fastSort(nums, hi+1, end);
    }
    private void swap(int[] nums, int p, int q) {
        int temp = nums[p];
        nums[p] = nums[q];
        nums[q] = temp;
    }
}