题目:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。示例:给定 nums = [2, 7, 11, 15], target = 9
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解决方案
方法一:
通常看到这题,第一反应就是暴力法。遍历每个元素 x,并查找是否存在一个值与 target - x 相等的目标元素,需要两层for循环。
func twoSum(nums []int, target int) []int {
n := len(nums)
ret := []int{}
for i := 0; i < n-1; i++ {
for j := i + 1; j < n; j++ {
if nums[i]+nums[j] == target {
ret = append(ret, i, j)
return ret
}
}
}
return ret
}
复杂度分析:
- 时间复杂度:O(n^2)O(n2), 对于每个元素,我们试图通过遍历数组的其余部分来寻找它所对应的目标元素,这将耗费 O(n)O(n) 的时间。因此时间复杂度为 O(n^2)O(n2)。
- 空间复杂度:O(1)O(1)。
方法二:
一遍哈希表。
在进行迭代并将元素插入到表中的同时,我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在,那我们已经找到了对应解,并立即将其返回。
func twoSum(nums []int, target int) []int {
tables := make(map[int]int)
for index, value := range nums {
tmp := target - value
if i, ok := tables[tmp]; ok {
return []int{i, index}
}
tables[value] = index
}
return []int{}
}
复杂度分析:
- 时间复杂度:O(n), 我们只遍历了包含有 n 个元素的列表一次。在表中进行的每次查找只花费 O(1)O(1) 的时间。
- 空间复杂度:O(n), 所需的额外空间取决于哈希表中存储的元素数量,该表最多需要存储 n 个元素。