版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/swty3356667/article/details/84146735
问题描述:给定一个整数, 从里面找出两个数, 其和等于一个指定的整数. 程序返回这两个数在数组中的位置( 数组下标从1开始 ) , 且位置小的在前面.
问题分析:一开始想的定义两个指针p1和p2,分别指向数组的开始和结尾,num[p1] + num[p2]与key的大小,相等则返回,如果key小,p2--,否则p1++,循环条件是p1 < p2。但是现在有一种更好的方法就是用哈希表存储,这样只需要遍历一遍,拿到一个元素时,直接用key减,得到目标值,在哈希表中找这个目标值,时间复杂度为O(1),找到就直接返回这两个下标,找不到就继续遍历。
代码:
class Solution
{
public:
vector<int> twoSum(vector<int> &num,int key)
{
unordered_map<int,int> map;//哈希表
vector<int> result;
for(int i = 0; i < num.size();i++)//插入到容器map
{
map[num[i]] = i;//key为num[i],value为i
}
for(int i = 0; i < num.size();i++)//遍历容器
{
const int gap = key - num[i];
if(map.find(gap) != map.end() && map[gap] > i)
{
result.push_back(i + 1);
result.push_back(map[gap] + 1);
break;
}
}
return result;
}
};