两数之和

两数之和

 次点击
7 分钟阅读

难度:简单

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。

你可以假设每种输入只会对应一个答案,并且你不能使用两次相同的元素。

你可以按任意顺序返回答案。

示例 1:

输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

示例 2:

输入:nums = [3,2,4], target = 6
输出:[1,2]

示例 3:

输入:nums = [3,3], target = 6
输出:[0,1]

提示:

  • 2 <= nums.length <= 10^4

  • -10^9 <= nums[i] <= 10^9

  • -10^9 <= target <= 10^9

  • 只会存在一个有效答案

进阶: 你可以想出一个时间复杂度小于 O(n^2) 的算法吗?

思路

使用 map 的形式存储 k,v 对去减小时间复杂度

解题过程

每次遍历的时候查询 map 中是否有对应的 key 和 target-num 对应,若包含取出当前索引和 map 中索引返回,否则将当前数字和索引以 k,v 对存到 hashmap 中

复杂度

时间复杂度: O(n)

空间复杂度: O(n)

func twoSum(nums []int, target int) []int {
    hashmap := make(map[int]int, len(nums)-1)
    for i, num := range nums {
        if val, ok := hashmap[target-num]; ok {
            return []int{val, i}
        }
        hashmap[num] = i
    }
    panic("error")
}

© 本文著作权归作者所有,未经许可不得转载使用。