Two Sum 两数之和
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.
Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
思路分析
- 参数:一个数字数组nums,一个目标数target。
- 目标:让找到两个数字(同一个数字不能使用两次),使其和为target。
- 分析:暴力搜索的话是遍历所有的两个数字的组合计算其和,这样节省了空间,但是时间复杂度高。为了减少时间的复杂度,需要用空间来换。
只遍历一次数组,在遍历的同时将数据使用HashMap存储起来,翻转key和value,建立数字和其坐标位置之间的映射。HashMap 是常数级的查找效率,这样在遍历数组的时候,用 target 减去遍历到的数字,就是另一个需要的数字了。直接在 HashMap 中查找其是否存在即可。
由于当前数字在HashMap中还不存在,所以不会出现同一个数字使用两次的情况。比如 target 是4,遍历到了一个2,另外一个2还未进入HashMap中,肯定就是要找的值。
整个实现步骤为:遍历一遍数组,查找target 减去当前数字是否存在于HashMap中;如果不存在,则建立 HashMap 映射。代码如下:
Go代码
func twoSum(nums []int, target int) []int {
m := make(map[int]int)
for k, v := range nums {
idx, ok := m[target - v]
if ok {
return []int{idx, k}
}
m[v] = k
}
return nil
}
// nums := []int{2, 7, 11, 15}
// target := 9
// result [1 0]
网友评论