力扣刷题13

第一题:四数之和

来源:https://leetcode.cn/problems/4sum/description/

题目:

给你一个由 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

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

python 复制代码
from typing import List

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        nums.sort()
        n = len(nums)
        result = []
        
        for i in range(n - 3):
            # 去重第一个数
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            for j in range(i + 1, n - 2):
                # 去重第二个数
                if j > i + 1 and nums[j] == nums[j - 1]:
                    continue
                left = j + 1
                right = n - 1
                while left < right:
                    total = nums[i] + nums[j] + nums[left] + nums[right]
                    if total == target:
                        result.append([nums[i], nums[j], nums[left], nums[right]])
                        # 去重第三个数
                        while left < right and nums[left] == nums[left + 1]:
                            left += 1
                        # 去重第四个数
                        while left < right and nums[right] == nums[right - 1]:
                            right -= 1
                        left += 1
                        right -= 1
                    elif total < target:
                        left += 1
                    else:
                        right -= 1
        return result

这个题和前面的"三数之和"差不多,就是多了一个数字而已。

我的思路仍然是降维,将四数之和转换为两数之和。先对数组进行排序,先固定前两个数,将问题简化为在剩下的区间里找两数之和,使四数之和等于题目中所给的target,用双指针寻找后两个数,同时跳过重复元素,避免生成重复的四元组。

我用题目中所给的示例进行模拟验证:

示例:nums = [1,0,-1,0,-2,2],target = 0

步骤1:排序

原数组:[1,0,-1,0,-2,2]

排序后的数组:[-2,-1,0,0,1,2],len(nums) = 6

步骤2:第一层循环(i从0开始)

i = 0,nums[i] = -2,进入第二层循环,j从j + 1开始

j = 1,nums[j] = -1,初始化left = 2,right = 5,nums[2] = 0,nums[5] = 2,进入循环;

total = nums[i] + nums[j] + nums[left] + nums[right] = -2 - 1 + 0 + 2 = -1 < target,left += 1;

此时left = 3,nums[left] = 0,进入循环

total = nums[i] + nums[j] + nums[left] + nums[right] = - 2 - 1 + 0 + 2 = - 1 < target,left += 1;

此时left = 4,nums[left] = 1,进入循环

total = nums[i] + nums[j] + nums[left] + nums[right] = - 2 - 1 + 1 + 2 = 0 == target,加入result。

做去重检查:检查nums[left]和nums[left + 1],无重复,nums[right]和nums[right - 1],无重复;

left += 1,right -= 1,此时left = 5 > right,退出循环。

j = 2,nums[j] = 0

初始化left = 3,right = 5,nums[left] = 0,nums[right] = 2,进入循环;

total = nums[i] + nums[j] + nums[left] + nums[right] = - 2 + 0 + 0 + 2 = 0 == target,加入result;

做去重检查:检查nums[left]和nums[left + 1],无重复,nums[right]和nums[right - 1],无重复;

left += 1,right -= 1,此时left = 4,right = 4,退出循环。

j = 3,nums[j] = 0(此情况下也存在重复现象,可直接排除)

初始化left = 4,right = 5,nums[left] = 1,nums[right] = 2,进入循环;

total = nums[i] + nums[j] + nums[left] + nums[right] = - 2 + 0 + 1 + 2 = 1 > target,right -= 1;

此时right = 4,退出循环。

j = 4,nums[j] = 1

初始化left = 5,right = 5,退出循环;


i = 1,nums[i] = -1,进入第二层循环,j从j + 1开始

j = 2,nums[j] = 0,初始化left = 3,right = 5,nums[left] = 0,nums[right] = 2,进入循环;

total = nums[i] + nums[j] + nums[left] + nums[right] = - 1 + 0 + 0 + 2 = 1 > target,right -= 1;

此时right = 4,nums[right] = 1,进入循环

total = nums[i] + nums[j] + nums[left] + nums[right] = - 1 + 0 + 0 + 1 == target,加入result;

做去重检查:nums[left]和nums[left + 1],无重复,nums[right]和nums[right - 1],无重复;

left += 1,right -= 1,left = 4 > right,退出循环。


i = 2,nums[i] = 0,进入第二层循环,j从j + 1开始

j = 3,nums[j] = 0,初始化left = 4,right = 5,nums[left] = 1,nums[right] = 2,进入循环;

total = nums[i] + nums[j] + nums[left] + nums[right] = 0 + 0 + 1 + 2 = 3 > target,right -= 1;

此时right = 4,退出循环。


最终结果:[-1,0,0,1],[-2,0,0,2],[-2,-1,1,2],与示例一致。

记录完毕,收工。

相关推荐
夜天炫安全9 小时前
数据结构中所需的C语言基础
c语言·数据结构·算法
2301_7890156210 小时前
DS进阶:AVL树
开发语言·数据结构·c++·算法
zyq99101_113 小时前
优化二分查找:前缀和降复杂度
数据结构·python·蓝桥杯
qyzm13 小时前
天梯赛练习(3月13日)
开发语言·数据结构·python·算法·贪心算法
逆境不可逃14 小时前
LeetCode 热题 100 之 64. 最小路径和 5. 最长回文子串 1143. 最长公共子序列 72. 编辑距离
算法·leetcode·动态规划
CoderCodingNo14 小时前
【GESP】C++五级练习题 luogu-P1182 数列分段 Section II
开发语言·c++·算法
放下华子我只抽RuiKe514 小时前
机器学习全景指南-直觉篇——基于距离的 K-近邻 (KNN) 算法
人工智能·gpt·算法·机器学习·语言模型·chatgpt·ai编程
kisshuan1239614 小时前
[特殊字符]【深度学习】DA3METRIC-LARGE单目深度估计算法详解
人工智能·深度学习·算法
sali-tec14 小时前
C# 基于OpenCv的视觉工作流-章33-Blod分析
图像处理·人工智能·opencv·算法·计算机视觉
Eward-an15 小时前
LeetCode 239. 滑动窗口最大值(详细技术解析)
python·算法·leetcode