📑 题目:1. 两数之和

🚀 本题 LeetCode 传送门

题目大意

在数组中找到 2 个数之和等于给定值的数字,结果返回 2 个数字在数组中的下标。

解题思路

这道题最优的做法时间复杂度是 O(n)。

顺序扫描数组,对每一个元素,在 map 中找能组合给定值的另一半数字,如果找到了,直接返回 2 个数字的下标即可。如果找不到,就把这个数字存入 map 中,等待扫到“另一半”数字的时候,再取出来返回结果。

代码

  1. package leetcode
  2. func twoSum(nums []int, target int) []int {
  3. m := make(map[int]int)
  4. for i := 0; i < len(nums); i++ {
  5. another := target - nums[i]
  6. if _, ok := m[another]; ok {
  7. return []int{m[another], i}
  8. }
  9. m[nums[i]] = i
  10. }
  11. return nil
  12. }