题目描述:
题号:1
题目描述:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
解题思路:
思路一:暴力法
最容易想到的方法。用两个循环枚举每种可能的组合
时间复杂度:O(N^N)
空间复杂度:O(1)
C++代码:
// C++
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
for (int i = 0; i < nums.size(); i++) {
for (int j = i + 1; j < nums.size(); j++) {
if (nums[i] + nums[j] == target) {
return {i, j};
}
}
}
return {};
}
};
思路二:哈希表
使用哈希表来存储数组下标(索引)和数组元素的对应关系,然后从哈希表中寻找有没有 target - nums[i]
时间复杂度:O(N)
空间复杂度:O(1)
踩坑:对于每一个nums[i],我们首先查询哈希表中是否存在 target - nums[i] ,然后将 nums[i] 插入到哈希表中,即可保证不会让 nums[i] 和索引配置或者和它自己匹配。
踩坑代码--没有先查询哈希表中是否存在答案:
// C++
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> subHash;
// 将数值和索引的对应关系存入map
for(int i = 0; i < nums.size(); i++) {
subHash[nums[i]] = i;
}
for(int i = 0; i < nums.size(); i++) {
auto index = subHash.find(target - nums[i]);
if(index != subHash.end()) {
return {index->second, i};
}
}
return {};
}
};
正确的代码:
C++代码:
// C++
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> subHash;
// 将数值和索引的对应关系存入map
for(int i = 0; i < nums.size(); i++) {
auto index = subHash.find(target - nums[i]);
if(index != subHash.end()) {
return {index->second, i};
}
subHash[nums[i]] = i;
}
return {};
}
};
go代码:
// go
func twoSum(nums []int, target int) []int {
subHash := map[int]int{}
for i, num := range nums {
if index, ok := subHash[target - num]; ok {
return []int{index, i}
}
subHash[num] = i
}
return nil
}