一、题目
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
二、解题
需要使用到hash表,在swift中使用的是Dictionary,首先创建一个字典,遍历nums,查询target-num是否存在在字典中,如果不存在,将num作为key,index作为value存入字典中,继续遍历下一个,直到字典中出现key=target-num的组合,当前key对应的value和当前的index就是要找的两个数的位置。
三、代码实现
class Solution {
func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
var numberIndexDict = [Int:Int]()
// 遍历数组
for (index, num) in nums.enumerated() {
// 判断字典中是否存在key = target - num
guard let pairedIndex = numberIndexDict[target - num] else {
// 把自身放入hash表
numberIndexDict[num] = index
continue
}
// 配对的数已存在
return [pairedIndex, index]
}
return [-1,-1]
}
}
Demo地址:github