15. 三数之和

15. 三数之和

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        for(int k = 0; k < nums.length - 2; k++){
            if(nums[k] > 0) break;
            if(k > 0 && nums[k] == nums[k - 1]) continue;
            int i = k + 1, j = nums.length - 1;
            while(i < j){
                int sum = nums[i] + nums[j] + nums[k];
                if(sum < 0){
                    //sum < 0，i指针右移且跳过重复元素
                    while(i < j && nums[i] == nums[++i]);
                } else if (sum > 0) {
                    //sum > 0，j指针左移且跳过重复元素
                    while(i < j && nums[j] == nums[--j]);
                } else {
                    //sum == 0，添加结果且同时移动俩指针（跳过重复元素）
                    res.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[i], nums[j])));
                    while(i < j && nums[i] == nums[++i]);
                    while(i < j && nums[j] == nums[--j]);
                }
            }
        }
        return res;
    }
}