题目：1. 两数之和
给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target 的那两个整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1：
输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
示例 2：
输入：nums = [3,2,4], target = 6
输出：[1,2]
示例 3：
输入：nums = [3,3], target = 6
输出：[0,1]
提示：
-
2 <=
nums.length<= -
<=
nums[i]<= -
<=
target<= -
只会存在一个有效答案
进阶： 你可以想出一个时间复杂度小于 的算法吗？
解题思路
这道题最优的算法时间复杂度是。
顺序扫描数组，对每一个元素，在 map 中找能组合给定值的另一半数字，如果找到了，直接返回 2 个数字的下标即可。如果找不到，就把这个数字存入 map 中，等待扫到“另一半”数字的时候，再取出来返回结果。
实现代码
func twoSum(nums []int, target int) []int {
diff := make(map[int]int, 0)
for i := 0; i < len(nums); i++ {
idx, ok := diff[target-nums[i]]
if ok {
return []int{idx, i}
}
diff[nums[i]] = i
}
return []int{}
}
复杂度分析
-
时间复杂度: , 其中 是数组中的元素数量。对于每一个元素
x，我们可以 地寻找target - x。 -
空间复杂度: , 其中 是数组中的元素数量。主要为哈希表的开销。
单元测试
package leetcode
import (
"testing"
"github.com/stretchr/testify/assert"
)
func Test_twoSum(t *testing.T) {
assert := assert.New(t)
type args struct {
nums []int
target int
}
tests := []struct {
args args
want []int
}{
{
args: args{nums: []int{2, 7, 11, 15}, target: 9},
want: []int{0, 1},
},
{
args: args{nums: []int{3, 2, 4}, target: 6},
want: []int{1, 2},
},
{
args: args{nums: []int{3, 3}, target: 6},
want: []int{0, 1},
},
}
for _, tt := range tests {
actual := twoSum(tt.args.nums, tt.args.target)
assert.Equal(tt.want, actual)
}
}
