四数之和——python

2023-11-15

18. 四数之和

难度中等997

给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):

  • 0 <= a, b, c, d < n
  • abc 和 d 互不相同
  • nums[a] + nums[b] + nums[c] + nums[d] == target

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

示例 1:

输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2:

输入:nums = [2,2,2,2,2], target = 8
输出:[[2,2,2,2]]

提示:

  • 1 <= nums.length <= 200
  • -109 <= nums[i] <= 109
  • -109 <= target <= 109

我的解析思路:

有了前面的两数之和的基础,我首先想到的利用双指针的方法来解决这个问题;整个调试过程真的很艰难,最后在自己的努力下终于AC了(虽然时空复杂度并不是很优秀)。

调试过程主要要注意一下几点:

  • 双指针的前提要先排序
  • 题目要求同一个位置的元素相同的话就视为重复的四元组,所以在处理第一个元素和第二个元素的时候我分别利用f1和f2标记重复元素的个数,如果是相同元素的时候就结束本次循环,直到不是重复元素为止。这里需要注意的是:f2为了避免最外层的上一次的循环的标记给下一次的结果造成错误的影响,所以每次循环之前要对它设置为0,否则就会出错。
  • 对于每次的第三个元素(left)重复的处理要注意:一直移动到倒数第二个重复的值的位置,因为可能存在第四个元素(right)为这个重复值的情况。还要判断它有没有使用,如果没有使用,而是继续自己使用的情况,那就是重复四元组。
  • 同样的要处理第四个元素。
  • 对于找到目标值之后而不是跳出循环,而是继续移动第三个元素。因为可能存在中间的两个值相加等于外层两个数相加的情况,所以不能漏掉。

我的代码:

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        result_ls = []
        len_nums = len(nums)
        #线排序一下,这样就可使用双指针的方法
        nums.sort()
        #print(nums)
        f1 = 0
        f2 = 0
        for element1 in range(len_nums - 3):
            if f1 > 0:
                f1 -= 1
                continue
            f2 = 0
            for element2 in range(element1 + 1, len_nums - 2):
                left = element2 + 1
                right = len_nums - 1
                if f2 > 0:
                    f2 -= 1
                    continue
                while(left < right):

                    sum = nums[element1] + nums[element2] + nums[left] + nums[right]
                    #print(element1, element2, left, right)
                    if sum == target:
                        #print('resut',element1, element2, left, right)
                        t_ls = []
                        t_ls.append(nums[element1])
                        t_ls.append(nums[element2])
                        t_ls.append(nums[left])
                        t_ls.append(nums[right])
                        result_ls.append(t_ls)
                        left += 1
                        while left < right and nums[left] == nums[left - 1]:
                            #print('left', left)
                            left += 1
                        # 特殊情况就是留给right的一个值他并没有使用,而是继续自己使用的情况,那就是重复值
                        sum_new = nums[element1] + nums[element2] + nums[left] + nums[right]
                        if left + 1 == right and nums[left] == nums[left + 1] and (sum == target or sum_new > target):
                            break
                    elif sum < target:
                        left += 1
                        #print('-2',left,right)
                        while left + 1 < right and nums[left] == nums[left - 1]:
                            #print('left', left)
                            left += 1
                        # 特殊情况就是留给right的一个值他并没有使用,而是继续自己使用的情况,那就是重复值
                        sum_new = nums[element1] + nums[element2] + nums[left] + nums[right]
                        if left + 1 == right and nums[left] == nums[left + 1] and (sum == target or sum_new > target):
                            break
                    else:
                        right -= 1
                        #print('-1',left,right)
                        while left < right - 1 and nums[right] == nums[right + 1]:
                            #print('  ri -1')
                            right -= 1
                        sum_new = nums[element1] + nums[element2] + nums[left] + nums[right]
                        # 同样的,特殊情况就是留给left的一个值他并没有使用,而是继续自己使用的情况,那就是重复值
                        if left == right - 1 and nums[right] == nums[right - 1] and (sum == target or sum_new < target):
                            #print('le == ri -1')
                            break
                while element2 < left and nums[element2] == nums[element2 + 1]:
                    #print('ele2', element2)
                    element2 += 1
                    f2 += 1
            while element1 < len_nums - 3 and nums[element1] == nums[element1 + 1]:
                #print('ele1', element1)
                element1 += 1
                f1 += 1
        return result_ls

 运行结果:

参考答案:

分析:

四数之和,和15.三数之和是一个思路,都是使用双指针法, 基本解法就是在15.三数之和 的基础上再套一层for循环。

但是有一些细节需要注意,例如: 不要判断nums[k] > target 就返回了,三数之和 可以通过 nums[i] > 0 就返回了,因为 0 已经是确定的数了,四数之和这道题目 target是任意值。(大家亲自写代码就能感受出来)

15.三数之和的双指针解法是一层for循环num[i]为确定值,然后循环内有left和right下表作为双指针,找到nums[i] + nums[left] + nums[right] == 0。

四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值,依然是循环内有left和right下表作为双指针,找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况,三数之和的时间复杂度是O(n^2),四数之和的时间复杂度是O(n^3) 。

那么一样的道理,五数之和、六数之和等等都采用这种解法。

对于15.三数之和双指针法就是将原本暴力O(n^3)的解法,降为O(n^2)的解法,四数之和的双指针解法就是将原本暴力O(n^4)的解法,降为O(n^3)的解法。

之前我们讲过哈希表的经典题目:454.四数相加II,相对于本题简单很多,因为本题是要求在一个集合中找出四个数相加等于target,同时四元组不能重复。

454.四数相加II是四个独立的数组,只要找到A[i] + B[j] + C[k] + D[l] = 0就可以,不用考虑有重复的四个元素相加等于0的情况,所以相对于本题还是简单了不少!

我们来回顾一下,几道题目使用了双指针法。

双指针法将时间复杂度O(n^2)的解法优化为 O(n)的解法。也就是降一个数量级,题目如下:

操作链表:

双指针法在字符串题目中还有很多应用,后面还会介绍到。

 参考代码:

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        
        nums.sort()
        n = len(nums)
        res = []
        for i in range(n):
            if i > 0 and nums[i] == nums[i - 1]: continue
            for k in range(i+1, n):
                if k > i + 1 and nums[k] == nums[k-1]: continue
                p = k + 1
                q = n - 1

                while p < q:
                    if nums[i] + nums[k] + nums[p] + nums[q] > target: q -= 1
                    elif nums[i] + nums[k] + nums[p] + nums[q] < target: p += 1
                    else:
                        res.append([nums[i], nums[k], nums[p], nums[q]])
                        while p < q and nums[p] == nums[p + 1]: p += 1
                        while p < q and nums[q] == nums[q - 1]: q -= 1
                        p += 1
                        q -= 1
        return res

运行结果:

总结:

其实代码逻辑差别不大,但是明显参考答案更简洁,我可能存在下面一些原因:

1.对于双指针方法不够熟练

2.逻辑混乱导致觉着很复杂

但是通过调试的过程中加深了对双指针的理解,失败是成功之母!

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

四数之和——python 的相关文章

  • 如何手动计算分类交叉熵?

    当我手动计算二元交叉熵时 我应用 sigmoid 来获取概率 然后使用交叉熵公式并平均结果 logits tf constant 1 1 0 1 2 labels tf constant 0 0 1 1 1 probs tf nn sigm
  • 元组有什么用?

    我现在正在学习 Python 课程 我们刚刚介绍了元组作为数据类型之一 我阅读了它的维基百科页面 但是 我无法弄清楚这种数据类型在实践中会有什么用处 我可以提供一些需要一组不可变数字的示例吗 也许是在 Python 中 这与列表有何不同 每
  • 在 django ORM 中查询时如何将 char 转换为整数?

    最近开始使用 Django ORM 我想执行这个查询 select student id from students where student id like 97318 order by CAST student id as UNSIG
  • Pandas/Google BigQuery:架构不匹配导致上传失败

    我的谷歌表中的架构如下所示 price datetime DATETIME symbol STRING bid open FLOAT bid high FLOAT bid low FLOAT bid close FLOAT ask open
  • 处理 Python 行为测试框架中的异常

    我一直在考虑从鼻子转向行为测试 摩卡 柴等已经宠坏了我 到目前为止一切都很好 但除了以下之外 我似乎无法找出任何测试异常的方法 then It throws a KeyError exception def step impl contex
  • 独立滚动矩阵的行

    我有一个矩阵 准确地说 是 2d numpy ndarray A np array 4 0 0 1 2 3 0 0 5 我想滚动每一行A根据另一个数组中的滚动值独立地 r np array 2 0 1 也就是说 我想这样做 print np
  • 使用 xlrd 打开 BytesIO (xlsx)

    我正在使用 Django 需要读取上传的 xlsx 文件的工作表和单元格 使用 xlrd 应该可以 但因为文件必须保留在内存中并且可能不会保存到我不知道如何继续的位置 本例中的起点是一个带有上传输入和提交按钮的网页 提交后 文件被捕获req
  • Python,将函数的输出重定向到文件中

    我正在尝试将函数的输出存储到Python中的文件中 我想做的是这样的 def test print This is a Test file open Log a file write test file close 但是当我这样做时 我收到
  • 如何通过 TLS 1.2 运行 django runserver

    我正在本地 Mac OS X 机器上测试 Stripe 订单 我正在实现这段代码 stripe api key settings STRIPE SECRET order stripe Order create currency usd em
  • 如何使用 pybrain 黑盒优化训练神经网络来处理监督数据集?

    我玩了一下 pybrain 了解如何生成具有自定义架构的神经网络 并使用反向传播算法将它们训练为监督数据集 然而 我对优化算法以及任务 学习代理和环境的概念感到困惑 例如 我将如何实现一个神经网络 例如 1 以使用 pybrain 遗传算法
  • Numpy - 根据表示一维的坐标向量的条件替换数组中的值

    我有一个data多维数组 最后一个是距离 另一方面 我有距离向量r 例如 Data np ones 20 30 100 r np linspace 10 50 100 最后 我还有一个临界距离值列表 称为r0 使得 r0 shape Dat
  • 仅第一个加载的 Django 站点有效

    我最近向 stackoverflow 提交了一个问题 标题为使用mod wsgi在apache上多次请求后Django无限加载 https stackoverflow com questions 71705909 django infini
  • 如何使用原始 SQL 查询实现搜索功能

    我正在创建一个由 CS50 的网络系列指导的应用程序 这要求我仅使用原始 SQL 查询而不是 ORM 我正在尝试创建一个搜索功能 用户可以在其中查找存储在数据库中的书籍列表 我希望他们能够查询 书籍 表中的 ISBN 标题 作者列 目前 它
  • 如何在 Windows 命令行中使用参数运行 Python 脚本

    这是我的蟒蛇hello py script def hello a b print hello and that s your sum sum a b print sum import sys if name main hello sys
  • Pandas 将多行列数据帧转换为单行多列数据帧

    我的数据框如下 code df Car measurements Before After amb temp 30 268212 26 627491 engine temp 41 812730 39 254255 engine eff 15
  • 如何在 pygtk 中创建新信号

    我创建了一个 python 对象 但我想在它上面发送信号 我让它继承自 gobject GObject 但似乎没有任何方法可以在我的对象上创建新信号 您还可以在类定义中定义信号 class MyGObjectClass gobject GO
  • 在本地网络上运行 Bokeh 服务器

    我有一个简单的 Bokeh 应用程序 名为app py如下 contents of app py from bokeh client import push session from bokeh embed import server do
  • Django-tables2 列总计

    我正在尝试使用此总结列中的所有值文档 https github com bradleyayers django tables2 blob master docs pages column headers and footers rst 但页
  • 使用 z = f(x, y) 形式的 B 样条方法来拟合 z = f(x)

    作为一个潜在的解决方案这个问题 https stackoverflow com questions 76476327 how to avoid creating many binary switching variables in gekk
  • 使用随机放置的 NaN 创建示例 numpy 数组

    出于测试目的 我想创建一个M by Nnumpy 数组与c随机放置的 NaN import numpy as np M 10 N 5 c 15 A np random randn M N A mask np nan 我在创建时遇到问题mas

随机推荐