https://leetcode-cn.com/problems/two-sum/
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
最朴素的解法就是循环:
func twoSum(nums []int, target int) []int {
result := make([]int, 0, 2)
length := len(nums)
if length == 0 {
return result
}
for i := 0; i < length-1; i++ {
for j := i + 1; j < length; j++ {
if nums[i]+nums[j] == target {
result = append(result, i, j)
}
}
}
return result
}
这么样也可以通过,但是性能比较一般,时间复杂度为 O(n2)
我们也可以使用 map 来优化速度,这里需要注意,数组中元素可能是重复的,所以使用 map[int]int 没法满足需求,必须用 map[int][]int
代码如下:
func twoSum(nums []int, target int) []int {
m := make(map[int][]int)
for index, num := range nums {
m[num] = append(m[num], index)
}
for i, num := range nums {
left := target - num
fmt.Println(i, num, left, m[left])
if indexs, ok := m[left]; ok {
// 从 indexs 里面找出不等于 i 的位置
for _, index := range indexs {
if index != i {
return []int{i, index}
}
}
}
}
return []int{}
}
优化之后代码的性能提升巨大。
从 116ms -> 8ms 性能提升了几十倍,而且测试用例也全部通过。