【Golang】LeetCode 1. 两数之和

1. 两数之和

题目描述

思路

忙了一个月其他的事情，好久没刷题了，用经典的「1. 两数之和」找回一些手感。

这道题要求我们寻找数组当中的两个数，使得这两个数相加可以得到target，并且保证有效的答案只有一个，因此我们应该使用哈希表的方式来解决这个问题。

具体来说，我们建立一个map[int]int，用来记录数组中某个数出现时的下标。

我们采用一次遍历的方式，如果当前的数nums[i]满足j, ok := mp[target - nums[i]]，其中ok == true，则表示当前出现的数与数组当中之前出现过的某个数组成的和恰好为target，满足题意，直接返回[]int{i, j}，代表找到了答案。

在进行条件判断之后，如果没有找到答案，我们则应该记录一次当前出现的数的下标在哈希表当中。

虽然题目已经确保一定有答案，但是如果在函数结束之前不进行一次返回，Golang 的编译器将会报错，我们返回一个空数组来避免报错。需要注意的是，这个数组的类型需要与返回值指定的类型相同。

Golang 题解

func twoSum(nums []int, target int) []int {
    mp := make(map[int]int)
    n := len(nums)

    for i := 0; i < n; i ++ {
        if j, ok := mp[target - nums[i]]; ok {
            return []int{i, j}
        }

        mp[nums[i]] = i
    }

    return []int{}
}