LeetCode 2.1.7 Two Sum

版权声明:本文为博主原创文章,未经博主允许不得转载。 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;
	}
};

猜你喜欢

转载自blog.csdn.net/swty3356667/article/details/84146735