[LeetCode-Golang] 1. 两数之和

题目

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

 

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

 

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum

思路

1.创建map映射，记录nums[i]到i的映射
2.遍历目标数组，判断map中是否包含key为差值target-nums[i]的元素

3.如果包含，则返回map[target-nums[i]]与i；如果map中不包含，则插入map[nums[i]] = i

复杂度分析

时间复杂度 O(N)

空间复杂度 O(N)

代码

func twoSum(nums []int, target int) []int {
    // 创建map，记录nums[i]到i的映射
    m := make(map[int]int, len(nums))
    // 遍历nums
    for i, v := range nums {
        // 如果m[target-v]存在，则找到，返回两个索引
        if j, ok := m[target-v]; ok {
            return []int{j,i}
        } else {
            // 如果m[target-v]不存在，则添加m[v] = i
            m[v] = i
        }
    }
    return []int{-1,-1}
}