leetcode两数之和，三数之和，四数之和问题

最新推荐文章于 2023-07-06 21:12:17 发布

旺旺棒棒冰

最新推荐文章于 2023-07-06 21:12:17 发布

阅读量202

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode python基础文章标签： leetcode 两数之和三数之和四数之和双指针

本文链接：https://blog.csdn.net/ltochange/article/details/120154720

python基础同时被 2 个专栏收录

28 篇文章

订阅专栏

leetcode

18 篇文章

订阅专栏

本文介绍了如何使用双指针技巧优化Python实现的三数之和和四数之和问题，通过先排序数组并跳过重复元素，利用两个指针搜索满足和目标值的不重复子集。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 两数之和

在这里插入图片描述
遍历数组的同时，使用字典（哈希表）记录数对应的索引，对于每一个数nums[i]，判断 target-nums[i]是否在字典中，找到一个即返回。若列表中的与元素有重复也可这样处理，因为遍历的过程中，不包含当前元素。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        n =  len(nums)
        if n < 2:
            return []
        res_dict = {}
        for i in range(n):
            val = target - nums[i]
            if val in res_dict:
                idx = res_dict[val]
                return [i,idx]
            else:
                res_dict[nums[i]] = i
                # 存在相同元素，会覆盖。这个不影响结果
        return []

15. 三数之和

在这里插入图片描述

class Solution:
     def threeSum(self, nums):
        n = len(nums)
        if n < 3:
            return []

        nums.sort()
        res = []
        for i in range(n):
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            target = - nums[i]
            k = n - 1
            for j in range(i + 1, n):
                if j > i + 1 and nums[j] == nums[j - 1]:
                    continue
                while k > j and nums[k] + nums[j] > target:
                    k -= 1
                if k == j:
                    break
                if nums[k] + nums[j] == target:
                    tmp = [nums[i], nums[j], nums[k]]
                    res.append(tmp)
        return res

18. 四数之和

在这里插入图片描述

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        n = len(nums)
        if n < 4: return []

        nums.sort()
        res = []
        for i in range(n):
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            for j in range(i + 1, n):
                if j > i + 1 and nums[j] == nums[j - 1]:
                    continue
                val = target - nums[i] - nums[j]
                l = n - 1
                for k in range(j + 1, n):
                    if k > j + 1 and nums[k] == nums[k - 1]:
                        continue
                    while l > k and nums[l] + nums[k] > val:
                        l -= 1
                    if l == k:
                        break
                    if nums[l] + nums[k] == val:
                        tmp = [nums[i], nums[j], nums[k], nums[l]]
                        res.append(tmp)
        return res