本文始发于个人公众号:TechFlow

今天是周末,和大家一起来看一道算法题。这道题是大名鼎鼎的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.

翻译

给定一个全是int的数组和一个整数target,要求返回两个下标,使得数组当中这两个下标对应的和等于target。
你可以假设一定值存在一个答案,并且一个元素不能使用两次。

解答

找两个数和等于target,第一反应就是暴力枚举。假设数组长度是n,那么一个
n2的循环就可以搞定。用Python的话,分分钟就可以写出代码。

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

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

引入map

如果你熟悉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,并想到解法,并不容易,需要基于充分的思考和理解。

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

如果喜欢本文,请顺手给个关注吧:

LeetCode 1 Two Sum——在数组上遍历出花样相关推荐

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

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

  2. 遍历数组的方法和数组的遍历方法(3+10种)(上)

    文章目录 简介 遍历数组的方法 for循环遍历 特点: 使用: 输出: for in遍历 特点: 使用: 输出: 3. for of循环 特点: 使用: 输出: 简介 总所周知,遍历的方法有很多种,有 ...

  3. LeetCode第一阶段(一)【数组篇】

    LeetCode 283 Move Zeros 给定一个数组nums,写一个函数,将数组中所有的0挪到数组的末尾,而维持其他所有非0元素的相对位置. 举例:nums = [0,1,0,3,12],函数 ...

  4. leetcode 112. Path Sum, 113. Path Sum II | 112,113. 路径总和 I, II(Java)

    题目 https://leetcode.com/problems/path-sum/ https://leetcode.com/problems/path-sum-ii/ 题解 简单的遍历二叉树,不解 ...

  5. leetcode 304. Range Sum Query 2D - Immutable |304. 二维区域和检索 - 矩阵不可变(二维前缀和问题)

    题目 https://leetcode.com/problems/range-sum-query-2d-immutable/ 题解 本题是 medium 难度,二维前缀和问题.相似题目有: Easy: ...

  6. 【LeetCode】面试题 03. 数组中重复的数字

    [LeetCode]面试题 03. 数组中重复的数字 文章目录 [LeetCode]面试题 03. 数组中重复的数字 一.遍历数组 二.原地置换 总结 一.遍历数组 由于只需要找出数组中任意一个重复的 ...

  7. 数据科学 IPython 笔记本 9.5 NumPy 数组上的计算:通用函数

    9.5 NumPy 数组上的计算:通用函数 本节是<Python 数据科学手册>(Python Data Science Handbook)的摘录. 译者:飞龙 协议:CC BY-NC-S ...

  8. 【动态规划】LeetCode 377. Combination Sum IV

    LeetCode 377. Combination Sum IV Solution1: 我的未能AC的答案 题目描述的和前几道题差不多,但实际上不能用DFS来做(会超时),要用动态规划,还是记录一下吧 ...

  9. LeetCode刷题笔记2——数组2

    LeetCode刷题笔记2--数组2 重塑数组 题目 在 MATLAB 中,有一个非常有用的函数 reshape ,它可以将一个 m x n 矩阵重塑为另一个大小不同(r x c)的新矩阵,但保留其原 ...

最新文章

  1. 为什么不从算法上优化机器的“眼睛”,而是从“大脑”入手?
  2. 第九周项目实践3 利用二叉树遍历思想解决问题
  3. [云炬创业基础笔记]第二章创业者测试3
  4. Oracle ASM 翻译系列第十一弹:高级知识 Offline or drop?
  5. g menu i meun_长沙话读“这里”,到底是阁(gó)里还是该(gái)里
  6. jQuery判断当前点击的是第几个li的代码
  7. Python bytes 和 string 相互转换 - Python零基础入门教程
  8. JS void运算符
  9. MySql的语法规范
  10. liunx grep sed
  11. 经验分享 PDF如何旋转页面,替换页面
  12. 主板检测卡c5_用诊断卡 搞定主板故障
  13. 难得一见的数据库事务异常 Deadlock found when trying to get lock解决办法dao.DeadlockLoserDataAccessException怎么办
  14. 通过微信开发测试号进行微信登录
  15. python hook pc微信_一起来用python玩一波微信呀 | 防撤回, 好友分析, 聊天机器人~...
  16. rmd中无法打开链结r_十个超级好用的R语言编程技巧,一般人绝不知道!
  17. 极限的无穷小和无穷大
  18. Android插件化——加载其他APP页面
  19. 调取大华SDK获取IPC音视频流并得到H264帧
  20. 局域网计算机配置扫描系统,fly42局域网计算机配置检测系统

热门文章

  1. php 时间戳 星座,最精准最新十二星座日期表 时间表
  2. 20级计算机应用数学试题,计算机等级考试试题试卷及答案.doc
  3. mysql删除表中所有数据
  4. 逃避不一定躲得过,面对不一定最难过
  5. 网络安全工程师就业前景
  6. 【面经】华为车BU面经
  7. 共享单车蓝牙锁方案phy6222系列蓝牙芯片
  8. NFT自由人数字艺术藏品有收藏价值吗?
  9. 三星s20Android11怎么退回10,三星已经开始在GALAXY S20+上测试安卓11系统
  10. 人工智能之父马文・明斯基