美文网首页
Golang LeetCode - 1. Two Sum 两数之

Golang LeetCode - 1. Two Sum 两数之

作者: Avery_up | 来源:发表于2020-03-31 11:52 被阅读0次

    Two Sum 两数之和

    Given an array of integers, return indices of the two numbers such that they add up to a specific target.

    You may assume that each input would have exactly one solution, and you may not use the same element twice.

    Example:

    Given nums = [2, 7, 11, 15], target = 9,

    Because nums[0] + nums[1] = 2 + 7 = 9,
    return [0, 1].

    思路分析

    • 参数:一个数字数组nums,一个目标数target。
    • 目标:让找到两个数字(同一个数字不能使用两次),使其和为target。
    • 分析:暴力搜索的话是遍历所有的两个数字的组合计算其和,这样节省了空间,但是时间复杂度高。为了减少时间的复杂度,需要用空间来换。
      只遍历一次数组,在遍历的同时将数据使用HashMap存储起来,翻转key和value,建立数字和其坐标位置之间的映射。HashMap 是常数级的查找效率,这样在遍历数组的时候,用 target 减去遍历到的数字,就是另一个需要的数字了。直接在 HashMap 中查找其是否存在即可。
      由于当前数字在HashMap中还不存在,所以不会出现同一个数字使用两次的情况。比如 target 是4,遍历到了一个2,另外一个2还未进入HashMap中,肯定就是要找的值。
      整个实现步骤为:遍历一遍数组,查找target 减去当前数字是否存在于HashMap中;如果不存在,则建立 HashMap 映射。代码如下:

    Go代码

    func twoSum(nums []int, target int) []int {
        m := make(map[int]int)
        for k, v := range nums {
            idx, ok := m[target - v]
            if ok {
                return []int{idx, k}
            }
            m[v] = k
        }
        return nil
    }
    // nums := []int{2, 7, 11, 15}
    // target := 9
    // result [1 0]
    

    相关文章

      网友评论

          本文标题:Golang LeetCode - 1. Two Sum 两数之

          本文链接:https://www.haomeiwen.com/subject/igfwuhtx.html