2021 后端笔试准备 16(算法题:四数之和)

四数之和

预备知识:

最好先看一下前面写的二数之和 和 三数之和 的解法,再来看这个会比较连贯好理解一点。

题目描述:

输入一个长度大于4的,整数类型的数组,和一个target(目标和)。求出这个list中有多少种组合,每种组合应该包含四个数字,这四个数字的和是16,并返回这些组合包含的数字。

例子:

输入: array = [7, 6, 4, -1, 1, 2], target = 16

输出:[[7, 6, 4, -1], [7, 6,1, 2]]

解题思路:

这道题最简单的解题思路

用 四个 for 循环去遍历数组,求出所有的可能性,将结果对应位16的数组放入result数组中然后返回

暴力解法的时间复杂度是O(n^4),因为需要遍历四次数组。 


接下来介绍另一种解法:

这种解法的思路是:将四数之和降低成二数之和

例如:

在 [7, 6, 4, -1, 1, 2] 这里list中我们可以把这些元素分成两个为一组的组合,例如我们第一个循环拿到7,嵌套的循环循环剩下的数字6,4,-1,1,2。我们可以得到 [7, 6], [7, 4], [7, -1], [7, 1],[7, 2],这样题目就好像退化成了二数之和。

首先我们先按在二数之和里面的思路去做,先把第一次循环得出来的组合放入到一个hash table中(python中是用dictionary),key是一对组合的结果,value就是一对组合 {13:[7,6]} , 同时我们得考虑两个数加和等于13的可能性有很多,所以value的部分我们得用二维数组,也就是{13:[[7,6]]}。剩下的元素在hash table 中是 { 13:[[7, 6]], 11: [[7, 4]], 6: [[7, -1]], 8: [[7, 1]],9: [[7, 2]] }, 我们把这个字典的名字命名为allPairSums意思就是所有组合的和。

上面的思路简单总结一下:循环得出组合,组合结果和target做差值,查看差值是否在 字典 中,如果字典中存在对应的差值,就将两对组合拼接在一起返回,如果不存在则,将新的组合放进字典中。

接下来我们只需要继续遍历list,然后用target减去遍历出来的组合得到差值,然后查看差值是否在hash table中,例如,我们二次以 6 为第一个数去遍历剩下的元素,我们可以得到 [6,4],得到6 + 4 的结果 10,用16 - 10 = 6, 然后我们查看 字典 中有没有对应的和为6的组合,我们可以发现 [7, -1]是等于6的这时候我们就可以把[7, -1]拿出来和[6, 4]组成一个长度为4的数组[7, -1, 6, 4],然后加入到结果的数组里面。这样我们就得到了第一个四数之和。请注意:在我们返回结果的时候我们需要去循环整个value数组,因为我们的值是一个二维数组,里面可能又不止一个元素。

请注意第二个需要注意的点!!!

我们的hash table 里面的值有可能出现两次,

举个栗子:

在allPairSums = { 13: [[7, 6]], 3: [[4, -1]], 10: [[6, 4]], 6: [[7, -1]]}中, 我们可以看到,13 + 3 = 16, 10 + 6 =16 字典中的两个4个元素两两组合都是可以得出结果16的,但是他们两两合并之后实际上都是等于[7, 6, 4, -1],这样我们实际上执行了两计算,同时我们在加入结果的时候也需要做检查,导致了额外的计算。

所以我们该如何避免呢?

举个栗子:

我们在数组 [7, 6, 4, -1, 1, 2] 第一个数选取的是4,我们现在的 字典 是空的{},我们先向4后面的元素开始遍历,我们可以得到[4, -1],[4, 1],[4, 2] ,每当我们得到一个新的组合我们需要去check字典中的值,但是字典现在是空的,所以没有对应的值,但是我们不像之前一样,将字典中没有的值放进去,在4和4后面的值组成的组合我们不放进去 allPairSums 字典中。

然后我们开始遍历4之前的元素,也就是7 ,6,我们可以得到[7, 4] ,去字典中查看是否有对应值,没有就加入字典, 如果有就append到数组和后面。6也是一样,得到[6, 4],然后去字典中查看,如果没有就加入字典中。现在字典就变成了{ 11: [[7, 4]], 10: [[6,4]]},如果现在有一个组合是[3,7] 数组就会变成{ 11: [[7, 4]], 10: [[6,4],[3,7]]}

总结一下上面的思路:第一次被选择数字,和索引在它后面的数字组成的组合不加入 字典中,只有第一次被选择数字,和索引在它前面的数字组合会加入字典中。

这样做的好处:

1. 能够避免加入6,7 和 7,6到数组中,因为只有被选的数字前面的能够加入字典中,这样就是一个单向循环,所以不会出现7,6 的情况,7只能选择和7之前的数字做成组合,但是7之前没有元素,所以它就没有组合。直到6被选择,它前面的数字是7,所以能和7组成 [6, 7] 这个组合。

2. 如果我们在第一次循环的时候就将后面的数字放入进 字典中:如图所示,如果我们在第一次遍历的时候就将元素加入到 字典中,我们就会出现重复组合的问题

接下来我们看代码:

我会放两个版本,一个版本是有注释的,想看没有注释的请将页面拉到最低下。



有注释版本

时间复杂度:

平均时间复杂度是: O(n^2)

时间的复杂度是O(n^2)的原因是,在我们的算法里, array中的每个元素都要被遍历一次,这个时间复杂度是O(n)。在遍历每个元素的时候我们还需要遍历元素前面和后面的元素,这个时间复杂度也是O(n),剩下的append到结果数组的操作,和计算组合结果的时间操作都是常数时间,也就是O(1)时间。所以总共是O(n^2)

空间复杂度是:O(n^2)

我们需要将每个元素和另一个元素的组合加入进字典O(n),同时需要将结果append到结果数组中O(n),所以加起来是O(n^2)

时间复杂度最差的情况是 O(n^3)

大家可以看下图


下面是没有注释的版本

无注释版本



Reference:代码来自Algoexpert,以学习作为主要目的来分享的。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 218,284评论 6 506
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,115评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,614评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,671评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,699评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,562评论 1 305
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,309评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,223评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,668评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,859评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,981评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,705评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,310评论 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,904评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,023评论 1 270
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,146评论 3 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,933评论 2 355

推荐阅读更多精彩内容