力扣刷题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],与示例一致。

记录完毕,收工。

相关推荐
追随者永远是胜利者1 小时前
(LeetCode-Hot100)253. 会议室 II
java·算法·leetcode·go
Jason_Honey22 小时前
【平安Agent算法岗面试-二面】
人工智能·算法·面试
程序员酥皮蛋2 小时前
hot 100 第三十五题 35.二叉树的中序遍历
数据结构·算法·leetcode
追随者永远是胜利者2 小时前
(LeetCode-Hot100)207. 课程表
java·算法·leetcode·go
仰泳的熊猫3 小时前
题目1535:蓝桥杯算法提高VIP-最小乘积(提高型)
数据结构·c++·算法·蓝桥杯
那起舞的日子3 小时前
动态规划-Dynamic Programing-DP
算法·动态规划
闻缺陷则喜何志丹4 小时前
【前后缀分解】P9255 [PA 2022] Podwyżki|普及+
数据结构·c++·算法·前后缀分解
每天吃饭的羊4 小时前
时间复杂度
数据结构·算法·排序算法
yzx9910134 小时前
Python数据结构入门指南:从基础到实践
开发语言·数据结构·python
ValhallaCoder5 小时前
hot100-堆
数据结构·python·算法·