LeetCode_ThreeSum_Array_M

15. 三數之和

  先排序，然後用雙指針即可快速查找。注意數據的去重

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
        int size = nums.length;
        if (size == 0 || nums[0] > 0 || nums[size - 1] < 0) {
            return result;
        }

        for (int i = 0; i < size && nums[i] <= 0; i++) {
            if (i > 0 && nums[i] == nums[i - 1]) {  //跳過重複數據
                continue;
            }
            int l = i + 1;
            int r = size - 1;
            while (l < r) {
                int count = nums[i] + nums[l] + nums[r];
                if (count == 0) {
                    result.add(Arrays.asList(nums[i], nums[l], nums[r]));
                    while (l < r && nums[l] == nums[l + 1])  l++;    //跳過重複數據
                    while (l < r && nums[r] == nums[r - 1])  r--;
                    l++;
                    r--;
                } else if (count > 0) {
                    while (l < r && nums[r] == nums[r - 1])  r--;
                    r--;
                } else {
                    while (l < r && nums[l] == nums[l + 1])  l++;
                    l++;
                }
            }
        }
        return result;
    }
}