1. 两数之和

题目描述

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

示例

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

解法

暴力解法
func twoSum(nums []int, target int) []int {
var ret []int
for index1, num1 := range nums {
for index2, num2 := range nums {
if index1 != index2 && num1+num2 == target {
ret = []int{index2, index1}
return ret
}
}
}
return ret
}
  • 时间复杂度: O(n^2)
  • 空间复杂度: O(1)
一遍哈希表
func twoSum(nums []int, target int) []int {
var (
ret []int
numsMap map[int][]int
)
numsMap = make(map[int][]int, len(nums))
for index, num := range nums {
if _, ok := numsMap[num]; ok {
numsMap[num] = append(numsMap[num], index)
} else {
numsMap[num] = []int{index}
}
if indexList, ok := numsMap[target-num]; ok {
for _, i := range indexList {
if i != index {
ret = []int{i, index}
return ret
}
}
}
}
return ret
}
  • 时间复杂度: O(n)
  • 空间复杂度: O(n)

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum