leetcode 18.四数之和 Java

题目链接

https://leetcode-cn.com/problems/4sum/

描述

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：

答案中不可以包含重复的四元组。

示例

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

满足要求的四元组集合为：
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

代码

与三数之和非常相似，只是增加了一层遍历，如果没有做过，建议做一下。

用四个指针，i、j、m、n，i遍历元素，j从i+1遍历元素，m以及n指向j右侧的两端，进行双指针的移动。
可以根据每次循环的最大值最小值，进行一部分数据的排除。

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> ans = new LinkedList<>();
        Arrays.sort(nums);
        int len = nums.length;
        for (int i = 0; i < nums.length - 3; i++) {
            //去重
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            //如果此次循环的最小值大于target，结束循环
            if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) {
                break;
            }
            //如果此次循环的最大值小于target，进入下一次循环
            if (nums[i] + nums[len - 1] + nums[len - 2] + nums[len - 3] < target) {
                continue;
            }

            for (int j = i + 1; j < len - 2; j++) {
                //去重
                if (j > i + 1 && nums[j] == nums[j - 1]) {
                    continue;
                }

                int m = j + 1;
                int n = len - 1;

                while (m < n) {
                    int sum = nums[i] + nums[j] + nums[m] + nums[n];

                    if (sum == target) {
                        ans.add(Arrays.asList(nums[i], nums[j], nums[m], nums[n]));
                        
                        while (m < n && nums[m] == nums[m + 1]) {
                            m++;
                        }
                        while (m < n && nums[n] == nums[n - 1]) {
                            n--;
                        }
                        m++;
                        n--;
                    } else if (sum < target) {
                        m++;
                    } else {
                        n--;
                    }
                }
            }
        }

        return ans;
    }
}