python 四数之和多种解法

编程小白狼

发布于 2024-12-31 08:13:12

4500

代码可运行

文章被收录于专栏：编程小白狼编程小白狼

运行总次数：0

代码可运行

方法一：暴力枚举

暴力枚举方法比较容易想到，就是将四个数的组合进行遍历，找到符合要求的组合。

代码如下：

def fourSum(nums, target):
    length = len(nums)
    nums.sort()
    result = []
    for i in range(length - 3):
        for j in range(i + 1, length - 2):
            for k in range(j + 1, length - 1):
                for l in range(k + 1, length):
                    if nums[i] + nums[j] + nums[k] + nums[l] == target:
                        if [nums[i], nums[j], nums[k], nums[l]] not in result:
                            result.append([nums[i], nums[j], nums[k], nums[l]])
    return result

时间复杂度为 O(n4)。

方法二：双指针

双指针方法比较高效，可以将时间复杂度降至O(n3)。

首先对数组进行排序，然后固定两个数，再用双指针法找到另外两个数，使其和等于目标值。

代码如下：

def fourSum(nums, target):
    nums.sort()
    n = len(nums)
    res = []
    for i in range(n-3):
        # 跳过重复元素
        if i > 0 and nums[i] == nums[i-1]:
            continue
        # 当前最小值大于目标值或者当前最大值小于目标值，直接退出
        if nums[i] + nums[i+1] + nums[i+2] + nums[i+3] > target:
            break
        if nums[i] + nums[n-1] + nums[n-2] + nums[n-3] < target:
            continue
        for j in range(i+1, n-2):
            # 跳过重复元素
            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, right = j+1, n-1
            while left < right:
                total = nums[i] + nums[j] + nums[left] + nums[right]
                if total == 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 total < target:
                    left += 1
                else:
                    right -= 1
    return res

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2023-12-28，如有侵权请联系 cloudcommunity@tencent.com 删除

数组