LeetCode刷题实战1:在数组上遍历出花样

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !

废话不多说,让我们一起来看看题目吧。

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

https://leetcode.com/problems/two-sum/

翻译

给定一个全是int的数组和一个整数target,要求返回两个下标,使得数组当中这两个下标对应的和等于target。

你可以假设一定值存在一个答案,并且一个元素不能使用两次。

解答

找两个数和等于target,第一反应就是暴力枚举。假设数组长度是n,那么一个

双重循环就可以搞定。用Python的话,分分钟就可以写出代码。

for i in range(len(array)):
    for j in range(len(array)):
        if array[i] + array[j] == target:
            return [i, j]

这样做当然是正确的,但显然不是最好的答案。根据经验,一般情况下O(n2)的算法都不是最优解。

如果你熟悉C++ STL或者其他语言工具库的用法,想必应该很容易想到可以使用map。map是一个非常常用的容器,用来存储key-value格式的数据对。在这道题当中,我们可以将元素和它在数组当中对应的下标存储进map当中。也就是说我们把所有数据对应的下标存储好了之后,我们在遍历的时候,就可以去掉j那一重循环,而直接判断target−a[j]在不在map当中即可。

我们继续写出伪代码:

for i in range(len(array)):
    map[array[i]] = i;
    if target - array[i] in map:
        return [i, map[target - array[i]]]

这个算法看起来没什么问题,但是如果你这么写出代码来提交一定过不了。

因为有一种隐藏的情况没有考虑到,一般我们会把这种隐藏的不容易想到的情况称作“Trick”,可以看做是出题人使用的诡计。

题目当中说了,同一个元素不能使用两次,但是并没有说数组当中没有重复的元素。map的使用有一个限制,就是不能有key值相同的元素。如果数组当中存在重复的元素,那么后面读到的数据会覆盖前面的。覆盖会产生什么问题呢?会导致我们没有办法判断元素出现的次数。

举个例子,比如:target=6, array=[3, 3]

由于我们使用了map,我们在记录下第二个3的时候,就会损失第一个3的信息。这样我们就会错过答案,不过这个问题也并不是不能解决,我们可以用一个标记记录一下,是否有重复的数字或者是重复的数字是什么。不过这并不是完美的解决方案,我们没有想到完美解法,是因为有一个潜在的条件被我们忽略了。

这个条件就是加法的交换律,也就是说a+b=target,那么b+a也应该等于target。这当然是一个废话,但如果a和b之间存在顺序的话就不一样了。如果a的顺序在b前面,那么我们应该通过a去找b呢,还是应该通过b去找a?

当然是通过b去找a,因为a在b之前,我们可以利用之前已经遍历过的信息查找。如果通过a去找b,那么我们需要再开辟一个循环遍历。

想明白了这点,剩下的就简单了。

假设数组array一共有n个元素,分别是a0,a1,⋯,an−1。我们用一个map存储之前出现过的元素的下标,当我们遍历到i的时候,我们只需要判断target−ai在不在map当中即可。

因为假设存在ai+aj=target,i<j。当我们指针遍历到i的时候找不出答案,因为aj的值还没有出现在map中。但是当我们遍历到j的时候,一定可以找到答案,因为ai已经出现过了。

通过利用加法交换律以及元素出现的先后顺序,再结合map,我们只需要一次遍历就可以找到答案。就算出现重复元素,也没有关系,因为我们是先判断存不存在,再更新map。

最后,附上伪代码:

for i in array.length:
    if target - array[i] in map:
        return i, map[target - array[i]]
    else:
        map.put(array[i], i)

这题本身并不难解,用到的知识也平淡无奇,不过要想能琢磨出其中的trick,并想到解法,并不容易,需要基于充分的思考和理解。

简单题也能有所收获,祝大家刷题愉快。

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