题目:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
解题思路&代码实现
暴力解法如下:循环两遍数组,每个元素和其他元素相加计算是否等于目标值,返回对应的下标。算法的时间复杂度是O(n^2)
package main
import (
"fmt"
)
func main() {
nums := []int{1, 3, 4, 6, 7, 10}
target := 17
newArr := twoSum(nums, target)
fmt.Println(newArr)
}
func twoSum(nums []int, target int) []int {
for key1, value1 := range nums {
for key2, value2 := range nums {
if key1 != key2 && value2 == target - value1 {
return []int{key1, key2}
}
}
}
return []int{}
}
优化算法:只循环一次,每次循环都以数组中的值作为索引,存储下标作为value。时间复杂度为O(n),并且每个元素只用一次。
package main
import (
"fmt"
)
func main() {
nums := []int{ 10,7,1, 3, 4, 6}
target := 17
newArr := twoSum(nums, target)
fmt.Println(newArr)
}
func twoSum(nums []int, target int) []int {
numsMap := make(map[int]int,0)
for key1, value1 := range nums {
value2 := target - value1
if value3 ,ok := numsMap[value2];ok{
return []int{key1, value3}
}
numsMap[value1] = key1
}
return []int{}
}