两数之和
1. 题目描述【简单】
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题思路
这道题目最开始大家可能的想法是用循环来做,使用循环也可以解决这样的问题,但是缺点就是效率太低。所以能不能有其他更好的算法来解决这个题呢?
2.1 使用map进行操作,算法流程参考图解LeetCode
注:图片来自五分钟学算法公众号
我们先定义几个变量:1. 目标数字target
2. 当前遍历到的数字now
3. 目标数字target
-当前数字now
的差值diff
。算法的流程首先是遍历数组,在遍历的过程中做了以下操作
- 判断
map
中的key
中是否含有diff
- 若包含,则返回
diff
对应的value
, 和now
对应的下标 - 不包含则将
now
放入map
中,其中key
为now
的值,value
为now
对应的下标
3. 代码实现
func twoSum(nums []int, target int) []int {
m := make(map[int]int)
for i, v := range nums {
if j, ok := m[target - v]; ok && i != j {
return []int{i, j}
}
m[v] = i
}
return []int{}
}