题目
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路
- 用双重for循环来做,但是时间复杂度是o(n²),很显然不是一个好的解题思路。
- 使用哈希映射来做,一次遍历,以值作为key,下标作为value,当遍历到一个值后,用目标值减去当前值作为key到哈希里面去找,找到就返回,没找到存入哈希里面。
代码 (swift)
func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
var hash = [Int: Int]()
for (index, num) in nums.enumerated() {
if let index1 = hash[target - num] {
return [index1, index]
} else {
hash[num] = index
}
}
return []
}