LeetCode 15. 三数之和（Medium）/ 18. 四数之和（Medium）/ 数字求和问题！！！

Aiclin

已于 2024-04-22 02:07:46 修改

阅读量155

点赞数

分类专栏： LeetCode 文章标签： leetcode

于 2020-12-18 20:59:42 首次发布

本文链接：https://blog.csdn.net/newCraftsman/article/details/111301917

版权

LeetCode 专栏收录该内容

100 篇文章 1 订阅

订阅专栏

题解

三数之和（排序+双指针，易懂图解）

思路

在这里插入图片描述

代码

class Solution:
    ### 1216 排序 + 遍历（900 ms，17.4 MB）
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = []

        # 首先对原数组排序
        nums.sort()

        # 从左到右固定每一个数（最后两个数无需遍历）
        for k in range(len(nums)-2):
            # 若当前nums[k]大于0，则后续无解，停止遍历
            if nums[k] > 0: break
            # 若有相同的k，则循环跳过
            if k > 0 and nums[k] == nums[k-1]: continue

            # 初始化双指针i、j
            i, j = k+1, len(nums)-1
            
            # 分类讨论
            while i < j:
                # 计算当前和s
                s = nums[k] + nums[i] + nums[j]

                # 若总和s小于0，表示s少了，则左指针i前移
                if s < 0:
                    i += 1
                    # 若有相同的i，则循环跳过
                    while i < j and nums[i] == nums[i-1]: i += 1
                # 若总和s大于0，表示s多了，则右指针j后移
                elif s > 0:
                    j -= 1
                    # 若有相同的j，则循环跳过
                    while i < j and nums[j] == nums[j+1]: j -= 1
                # 若总和s等于0，则记录当前的三元组，并移动左、右双指针
                else:
                    res.append([nums[k], nums[i], nums[j]])
                    
                    i += 1
                    j -= 1
                    while i < j and nums[i] == nums[i-1]: i += 1
                    while i < j and nums[j] == nums[j+1]: j -= 1
            
        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-3):
            # 防止重复 数组进入 res
            if i > 0 and nums[i] == nums[i-1]:
                continue
            # 当数组最小值和都大于target 跳出
            if nums[i] + nums[i+1] + nums[i+2] + nums[i+3] > target:
                break
            # 当数组最大值和都小于target,说明i这个数还是太小,遍历下一个
            if nums[i] + nums[n-1] + nums[n-2] + nums[n-3] < target:
                continue
            for j in range(i+1,n-2):
                # 防止重复 数组进入 res
                if j - i > 1 and nums[j] == nums[j-1]:
                    continue
                # 同理
                if nums[i] + nums[j] + nums[j+1] + nums[j+2] > target:
                    break
                # 同理
                if nums[i] + nums[j] + nums[n-1] + nums[n-2] < target:
                    continue
                # 双指针
                left = j + 1
                right = n - 1
                while left < right:
                    tmp = nums[i] + nums[j] + nums[left] + nums[right]
                    if tmp == target:
                        res.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 tmp > target:
                        right -= 1
                    else:
                        left += 1
        return res