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 <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
- 只会存在一个有效答案呢
思考:
本题是一道比较经典的哈希表的应用题,我们需要找的是相加为目标值的两个数,并且要输出这两个数的数组下标。所以我们需要记录的不仅是数组中数的值,还有数组中数的下标。此时,利用map这种数据结构尤为适合,map是一种key,value的存储结构,可以利用key保存数值,用value保存数组下标。
暴力求解:
但是显而易见的,这题可以很容易用两层for循环求解,暴力地将所有可能性都遍历一遍。
//暴力求解
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 { };
}
};
map哈希表:
因为对数组的排序并没有特殊要求,此处选择使用unordered_map哈希表, 效率更高。
//map哈希表
class Solution2 {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map <int, int> map; //定义map哈希表
for (int i = 0; i < nums.size(); i++)
{
if (map.find(target - nums[i]) != map.end())
return { map[target - nums[i]],i, };
map[nums[i]] = i;
}
return { };
}
};
C++中map,有三种类型:
映射 | 底层实现 | 是否有序 | 数值是否可以重复 | 能否更改数值 | 查询效率 | 增删效率 |
---|---|---|---|---|---|---|
std::map | 红黑树 | key有序 | key不可重复 | key不可修改 | O ( log n ) O(\log n) O(logn) | O ( log n ) O(\log n) O(logn) |
std::multimap | 红黑树 | key有序 | key可重复 | key不可修改 | O ( log n ) O(\log n) O(logn) | O ( log n ) O(\log n) O(logn) |
std::unordered_map | 哈希表 | key无序 | key不可重复 | key不可修改 | O ( 1 ) O(1) O(1) | O ( 1 ) O(1) O(1) |
双指针+哈希表:
当然,本题还可以使用双指针的方法检索对象,但是由于题意要求返回原本的数组下标,所以需要一个temp数组,防止下标在重新排序后丢失。
class Solution3 {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ans;
vector<int> temp;
temp = nums;
int n = temp.size();
sort(temp.begin(), temp.end());
int i = 0, j = n - 1;
while (i < j) {
if (temp[i] + temp[j] > target)j--;
else if (temp[i] + temp[j] < target)i++;
else break;
}
if (i < j) {
for (int k = 0; k < n; k++) {
if (i < n && nums[k] == temp[i]) {
ans.push_back(k);
i = n;
}
else if (j < n && nums[k] == temp[j]) {
ans.push_back(k);
j = n;
}
if (i == n && j == n)return ans;
}
}
return ans;
}
};
参考:
代码随想录
往期回顾:
LeetCode202. 快乐数
LeetCode350. 两个数组的交集 II
LeetCode349. 两个数组的交集
LeetCode1002. 查找共用字符