LeetCode题练习与总结:四数之和

194 篇文章 0 订阅
122 篇文章 0 订阅

一、题目描述

给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):

  • 0 <= a, b, c, d < n
  • abcd 互不相同
  • nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按 任意顺序 返回答案 。

示例 1:

输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2:

输入:nums = [2,2,2,2,2], target = 8
输出:[[2,2,2,2]]

提示:

  • 1 <= nums.length <= 200
  • -10^9 <= nums[i] <= 10^9
  • -10^9 <= target <= 10^9

二、解题思路

  1. 排序数组: 首先,我们需要对数组 nums 进行排序。排序是为了确保在寻找四元组时,我们可以利用有序数组的特性来减少不必要的比较。排序后,如果我们需要跳过重复的元素,可以更容易地实现。

  2. 外层循环: 外层循环从数组的第一个元素开始,遍历到倒数第四个元素。这是因为我们需要至少四个不同的元素来构成一个四元组,而且我们需要确保 abcd 互不相同。外层循环的变量 i 用于固定第一个元素。

  3. 跳过重复元素: 在每次外层循环时,我们需要检查当前元素是否与前一个元素相同。如果是,我们跳过当前循环,继续下一次循环,以避免重复的四元组。

  4. 内层循环: 内层循环从外层循环的下一个元素开始,遍历到倒数第三个元素。这个循环的变量 j 用于固定第二个元素。

  5. 双指针技巧: 对于每一对固定的 (i, j),我们使用两个指针 leftright 来遍历数组的剩余部分。left 指向 j + 1right 指向数组的最后一个元素。这两个指针分别从数组的两端向中间移动,尝试找到满足条件的 nums[left] + nums[right]

  6. 寻找四元组: 在双指针循环中,我们计算当前的四元组和 nums[i] + nums[j] + nums[left] + nums[right]。如果这个和等于目标值 target,我们就找到了一个有效的四元组。在添加到结果列表之前,我们需要跳过重复的元素,这通过在找到有效四元组后移动 leftright 指针来实现。

  7. 处理和值不匹配的情况: 如果当前四元组的和小于 target,我们需要增加 left 指针(因为增加较小的数会使和增大)。如果和大于 target,我们需要减小 right 指针(因为减小较大的数会使和减小)。

  8. 返回结果: 当所有的循环结束后,我们返回包含所有找到的不重复四元组的列表。

三、具体代码

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

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);

        // 遍历数组,对于每一个元素,使用双指针方法来找到另外两个元素的组合
        for (int i = 0; i < nums.length - 3; i++) {
            // 如果当前元素与前一个元素相同,跳过以避免重复
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }

            // 对于当前元素,使用双指针来找到另外两个元素
            for (int j = i + 1; j < nums.length - 2; j++) {
                // 如果当前元素与前一个元素相同,跳过
                if (j > i + 1 && nums[j] == nums[j - 1]) {
                    continue;
                }

                // 初始化左右指针
                int left = j + 1;
                int right = nums.length - 1;

                // 使用双指针找到满足条件的两个元素
                while (left < right) {
                    int sum = nums[i] + nums[j] + nums[left] + nums[right];
                    if (sum == target) {
                        // 找到满足条件的四元组,添加到结果中
                        result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                        // 移动指针,跳过重复元素
                        while (left < right && nums[left] == nums[left + 1]) {
                            left++;
                        }
                        while (left < right && nums[right] == nums[right - 1]) {
                            right--;
                        }
                        left++;
                        right--;
                    } else if (sum < target) {
                        // 如果和小于目标值,移动左指针
                        left++;
                    } else {
                        // 如果和大于目标值,移动右指针
                        right--;
                    }
                }
            }
        }

        return result;
    }
}

四、时间复杂度和空间复杂度

1. 时间复杂度
  • 数组排序的时间复杂度是 O(n log n),其中 n 是数组的长度。
  • 外层循环(i)遍历到数组的倒数第四个元素,因此有 n-3 次迭代。
  • 内层循环(j)同样遍历到倒数第三个元素,也有 n-3 次迭代。
  • 对于每一对 (i, j),我们使用双指针(left 和 right)来遍历剩余的元素。在最坏的情况下,双指针循环可能会进行 n-3 次迭代(当 left 和 right 从数组的两端开始移动时)。

因此,总的时间复杂度可以近似为 O(n log n + (n-3)^2 * (n-3))。由于 (n-3)^2 * (n-3) 相对于 n log n 是主导项,所以时间复杂度可以简化为 O(n^3)。

2. 空间复杂度
  • 我们使用了一个 ArrayList 来存储结果,其空间复杂度为 O(m),其中 m 是找到的有效四元组的数量。在最坏的情况下,m 可以是 n^4(当数组中的所有元素都可以组成四元组且满足条件时),但实际上 m 会远小于 n^4,因为不是所有的四元组都满足条件。
  • 我们还使用了一个临时的 List<Integer> 来存储单个四元组,这个 List 的空间复杂度是 O(1),因为它的大小是固定的(4个元素)。
  • 双指针循环中的变量(left 和 right)占用的空间是常数级别的。

综上所述,空间复杂度主要是由结果列表决定的,可以近似为 O(m),其中 m 是有效四元组的数量。在实际应用中,m 的值通常远小于 n^4,所以空间复杂度可以认为是 O(1)(常数级别)。

请注意,这里的分析是基于最坏情况的假设。在实际应用中,由于数组中可能存在大量不满足条件的元素,实际的时间和空间复杂度可能会低于上述分析。

五、总结知识点

  1. 数组排序:使用 Arrays.sort(nums) 方法对整数数组 nums 进行排序。这是解决这个问题的基础,因为排序后的数组可以让我们更容易地应用双指针技巧。
  2. 双指针技巧:在内层循环中,使用两个指针 leftright 来遍历数组的剩余部分,寻找满足特定条件的元素组合。这是一种常见的算法技巧,用于在有序数组中高效地找到特定值或满足特定条件的元素。
  3. 跳过重复元素:在外层循环和内层循环中,通过检查当前元素是否与前一个元素相同来跳过重复元素,以确保结果中的四元组元素互不相同。
  4. 条件判断与循环控制:在双指针循环中,通过计算当前四元组的和与目标值 target 的比较结果来决定移动哪个指针。如果和小于目标值,移动左指针;如果和大于目标值,移动右指针;如果和等于目标值,则找到有效四元组并添加到结果列表。
  5. 结果列表的构建:使用 ArrayList 来存储找到的四元组。在找到满足条件的四元组时,使用 Arrays.asList() 方法将数组元素转换为列表,并添加到结果列表中。
  6. 边界条件处理:在方法开始时,检查输入数组是否为 null 或长度小于 4,如果是,则直接返回空列表,因为无法构成四元组。

以上就是解决这个问题的详细步骤,希望能够为各位提供启发和帮助。

  • 61
    点赞
  • 35
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一直学习永不止步

谢谢您的鼓励,我会再接再厉的!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值