给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
主要思路:空间换时间,hashmap通过键搜索,时间复杂度为O(1);遍历数组,用一个字典存储数组的值和下标,键为元素值,值为下标。每一次遍历,算出target与当前元素的差值,用差值去hashmap中搜索,如果存在,即找到结果,拿出对应value(也即元素对应的下标),结果就是当前元素的下标与上一步取出的value值。
class Solution {
func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
var hashMap: [Int: Int] = [:]
var indexs = [Int]()
for num in nums.enumerated() {
let minus = target - num.element
if let index = hashMap[minus] {
indexs.append(index)
indexs.append(num.offset)
return indexs
}
hashMap[num.element] = num.offset
}
return []
}
}
网友评论