四数之和 | 算法

题目

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

思路

四数之和与之前的三数之和、两数之和存在共通性，即：都要确定左边界和右边界

在此题中，首先设置第一个数a在数组的最左端，第二个数b在a+1位置，第三个数left在b+1位置，第四个数right在nums.length-1的位置。

若left小于right，通过left和right相互逼近获取符合条件的数组，在获取到数组后，再判断nums[left] == nums[left+1]判断当前元素与下一个元素的值是否相等，相等需要跳过，最后返回整个结果集。

解题

    public List<List<Integer>> fourSum(int[] nums, int target) {
        // 判断是否是合法数组
        int size = nums.length;
        if (size < 4) {
            return Collections.emptyList();
        }
        // 排序
        Arrays.sort(nums);
        // 定义List接受结果
        List<List<Integer>> lists = new ArrayList<>();
        // 最外层循环
        for (int a = 0; a < size; a++) {
            // 如果最左边的数已经向右移动了，就判断和之前的数是否一样，若一样就跳出此次循环（获取第一个数）
            if (a > 0 && nums[a] == nums[a - 1]) {
                continue;
            }
            // 第二个数是在第一个数的基础上加1（获取第二个数）
            for (int b = a + 1; b < size; b++) {
                if (b > a + 1 && nums[b] == nums[b - 1]) {
                    continue;
                }
                // 定义第三个数（获取第三个数）
                int left = b + 1;
                // 把最后一个数放在数组的尾端(获取第四个数)
                int right = size - 1;
                // 若左指正小于右指针，执行循环
                while (left < right) {
                    // 若四数之和小于target，执行left++ 使target变大
                    if (nums[a] + nums[b] + nums[left] + nums[right] < target) {
                        left++;
                        // 若四叔之和大于target，执行right++，是target变小
                    } else if (nums[a] + nums[b] + nums[left] + nums[right] > target) {
                        right--;
                    } else {
                        // 四叔之和等于target 放入list中
                        lists.add(Arrays.asList(nums[a], nums[b], nums[left], nums[right]));

                        // 判断left将要右移的数是不是等于当前的值，若等于，执行left++
                        while (left < right && nums[left] == nums[left + 1]) {
                            left++;
                        }
                        // 判断right将要右移的数是不是等于当前的值，若等于，执行right--
                        while (left < right && nums[right] == nums[right - 1]) {
                            right--;
                        }
                        // left++ 继续执行下一次探索
                        left++;
                        // left++ 继续执行下一次探索
                        right--;
                    }
                }
            }
        }
        // 返回最后结果
        return lists;
    }

总结

有了两数之和和三数之和的经验，解决此题的思路就比较容易，但在自己写代码的过程中，还是会出现bug，还是一些小细节没有考虑周全

阅读原文