LeetCode-1

两数之和

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

哈希map保存数组元素与下标

class Solution 
{
public:
    vector<int> twoSum(vector<int>& nums, int target) 
    {
		int num = nums.size();
		unordered_map<int, int> hmap;
		vector<int> res(2, 0);
	
		for (int i = 0; i < num; i++)
		{
			int temp = target - nums[i];
			if (hmap.find(temp) != hmap.end())
			{
				res[0] = hmap[temp];
				res[1] = i;
			}		
			hmap[nums[i]] = i;
		}
		return res;
    }
};

count 和效率比find高

class Solution 
{
public:
    vector<int> twoSum(vector<int>& nums, int target) 
    {
	int num = nums.size();
	unordered_map<int, int> hmap;
	vector<int> res(2, 0);
	for (int i = 0; i < num; i++)
	{
		if (hmap.count(target - nums[i]) > 0)
		{
			res[0] = hmap[target - nums[i]];
			res[1] = i;
		}		
		hmap[nums[i]] = i;
	}
	return res;
};