[leetcode] 18. 四數之和

18. 四數之和

和以前的三數之和思路徹底同樣，只不過須要先枚舉兩個數，而後採用貪心的雙指針法。

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        int p, q;
        int len = nums.length;
        if (len < 4) {
            return new ArrayList<>();
        }
        int less = nums[0] + nums[1] + nums[2] + nums[3];
        if (less > target) {
            return new ArrayList<>();
        }
        int more = nums[len - 1] + nums[len - 2] + nums[len - 3] + nums[len - 4];
        if (more < target) {
            return new ArrayList<>();
        }

        List<List<Integer>> ans = new ArrayList<>();

        for (p = 0; p <= len - 4; p++) {
            // 對於重複出現的數字，枚舉一次就夠了
            if (p > 0 && nums[p] == nums[p - 1]) continue;
            for (q = p + 1; q <= len - 3; q++) {
                // 對於重複出現的數字，枚舉一次就夠了
                if (q > p + 1 && nums[q] == nums[q - 1]) continue;
                int i = q + 1, j = len - 1;
                while (i < j) {
                    int tmp = nums[p] + nums[q] + nums[i] + nums[j];
                    if (tmp == target) {
                        ans.add(Arrays.asList(nums[p], nums[q], nums[i], nums[j]));
                        i++;
                        j--;
                        // 對於重複出現的數字，枚舉一次就夠了;這兩個while保證不會有重複的結果
                        while (i < j && nums[i] == nums[i - 1]) i++;
                        while (i < j && nums[j] == nums[j + 1]) j--;
                    } else if (tmp < target) {
                        i++;
                    } else {
                        j--;
                    }
                }
            }
        }

        //return ans.stream().distinct().collect(Collectors.toList());
        return ans;
    }
}