两数之和
2019-03-18 本文已影响0人
Jabir_Zhang
题目:两数之和
描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
方法一:暴力法
暴力法很简单。遍历每个元素 xx,并查找是否存在一个值与 target - x相等的目标元素。
func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
let count = nums.count
for i in 0..<count {
for j in (i+1)..<count {
if nums[i] + nums[j] == target {
return [i,j]
}
}
}
return []
}
方法二:两遍哈希表
使用字典,时间复杂度O(n),空间复杂度O(n)
func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
let count = nums.count
var dic = [Int : Int]()
for i in 0..<count {
dic[nums[i]] = i
}
for i in 0..<count {
let found = target - nums[i]
if let j = dic[found], i != j {
return [i, j]
}
}
return []
}
方法三:一遍哈希表
方法二的优化
func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
let count = nums.count
var dic = [Int : Int]()
for i in 0..<count{
let found = target - nums[i]
if let j = dic[found] , j != i {
return [i,j]
}
dic[nums[i]]=i
}
return []
}