leetcode:两数之和

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

 

#include<iostream>
#include<vector>
#include<unordered_map>

using namespace std;

// 时间复杂度O(n)
// 空间复杂度O(n)
class Solution 
{
public:
	vector<int> twoSum(vector<int>& nums, int target) {
		unordered_map<int, int> record;
		for (int i = 0; i < nums.size(); i++)
		{
			int complment = target - nums[i];  //另一个目标值
			if (record.find(complment)!=record.end())
			{
				int res[2] = { i, record[complment] }; // 存储下标
				return vector<int>(res, res + 2);
			}
			record[nums[i]] = i;		// 没有找到另一个可加值，则添加元素和其角标
		}
		throw invalid_argument("the input has no solution");
	}
};

int main()
{
	vector<int> nums = { 1, 2, 3, 4, 6, 8, 9 };
	int target = 5;
	vector<int> result = Solution().twoSum(nums, 5);
	vector<int>::iterator iter;
	cout << "[";
	for ( iter = result.begin(); iter != result.end(); iter++)
	{
		if (iter == result.end()-1)
			cout << (*iter);
		else
		{
			cout << (*iter) << ",";
		}
	}
	cout << "]";


	system("pause");
	return 0;
}