每日一题——四数之和

菜鸡每日一题系列打卡18天

每天一道算法题目 

小伙伴们一起留言打卡

坚持就是胜利，我们一起努力！

题目描述(引自LeetCode)

给定一个包含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]
]
注意：答案中不可以包含重复的四元组。

题目分析

本题是一道四数之和的题目，做过之前三数之和题目的小伙伴解决这道题目应该能得心应手，其实思想是相同的，只不过这次需要用到四个指针。值得一提的是，可以使用剪枝来优化代码逻辑，提高程序运行速度。另外，需要做好去重的工作，避免答案中出现重复的四元组。

代码实现

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {


        // 用于记录结果
        List<List<Integer>> result = new ArrayList<>();


        // 特殊情况判断
        if (nums == null || nums.length < 4) {
            return result;
        }


        // 数组排序
        Arrays.sort(nums);


        // 获取数组长度
        int length = nums.length;


        for (int i = 0; i < length - 3; i++) {
            // 去重
            if (i > 0 && nums[i] == nums[i - 1]) continue;


            // 剪枝
            int min = nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3];
            if (min > target) break;
            int max = nums[length - 1] + nums[length - 2] + nums[length - 3] + nums[length - 4];
            if (max < target) break;


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


                int start = j + 1;
                int end = length - 1;


                // 剪枝
                int tmpMin = nums[i] + nums[j] + nums[start] + nums[start + 1];
                if (tmpMin > target) continue;
                int tmpMax = nums[i] + nums[j] + nums[end] + nums[end - 1];
                if (tmpMax < target) continue;
                
                while (start < end) {
                    // 计算当前值
                    int current = nums[i] + nums[j] + nums[start] + nums[end];


                    if (current == target) {
                        // 添加结果
                        result.add(Arrays.asList(nums[i], nums[j], nums[start], nums[end]));


                        // 去重
                        while (start < end && nums[start] == nums[start + 1]) start++;
                        while (start < end && j < end && nums[end] == nums[end - 1]) end--;


                        // 移动指针
                        start++;
                        end--;
                    } else if (current < target) {
                        // 向后移动start指针
                        start++;
                    } else {
                        // 向前移动end指针
                        end--;
                    }
                }
            }
        }
        // 返回结果
        return result;
    }
}

代码分析

对代码进行分析，首先对数组进行排序，时间复杂度是O(nlogn)，接下来针对每个数字进行四数之和的匹配，时间复杂度是O(n^3)，因此，整体的时间复杂度为O(n^3)。而就空间而言，在不考虑返回结果所占用空间的前提下，空间复杂度为O(1)。

执行结果

学习 | 工作 | 分享

????长按关注“有理想的菜鸡”

只有你想不到，没有你学不到