[LeetCode] 两数之和|刷题打卡

354 阅读1分钟

题目:

给定一个整数数组 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{}
}