leetcode 1. 两数之和

题目描述

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路

思路有两种，首先就是暴力遍历, $O(n^2)$。
二，取巧的方法

1. 将数组键值对反转存入map中，map[ val ] = key
2. 遍历数组， 假设此时遍历到 nums[i] = x;
3. 如果存在 x + y = target， 则map中索引y存在

代码如下

func twoSum(nums []int, target int) []int {

    m := make(map[int]int, len(nums)) //构建map

    for ikey,ival := range nums {
    //这里直接在一个循环中边存边查，相当于每次从当前索引向前查，避免了使用同一元素两次
        jval, ok := m[target - ival]
        if (ok) {
            return []int{jval,ikey} //颠倒下j,i的位置，小索引在前面
        }
        
        m[ival] = ikey
    }

    return []int{-1,-1}
}