题目难度: Easy
原题描述:
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
Example:
Given nums = [2, 7, 11, 15], target = 9, Because nums[0] + nums[1] = 2 + 7 = 9, return [0, 1].
题目大意:
给你一个包含n个数的数组和一个目标值,要你返回数组中和等于目标值的两个数的下标。本题假设每个输入都有一个唯一的解,而且结果的两个数不能是同一个元素。
解题思路:
对于数组中的每一个元素,使用map记录该元素对应的下标,即map的key为元素的值,value为元素的下标。记录的同时在map中查找key为(target-该元素的值)的元素是否存在,若存在且与原先的不是同一个元素,则返回这两个元素的下标;否则,则继续执行下一次循环。
时间复杂度分析:
由于使用了C++中的map,因此每次在在map中进行查找操作的复杂度为O(log(n)),总的复杂度为O(n*log(n))。而如果使用C++中的hash_map,则每次查找的复杂度为O(1),总的复杂度为O(n),但是由于LeetCode不支持C++的hash_map(自己加上#include <hash_map>也没有用),因此只能用map来做了。网上看见有人用C++11的unordered_map来做,时间复杂度也能去到O(n),其内部也是用哈希表来实现,也是很不错的方法。
以下是代码:
vector<int> twoSum(vector<int>& nums, int target)
{
int i;
map<int, int> m;
map<int, int>::iterator p;
vector<int> result;
for(i=0 ; i<nums.size() ; ++i)
{
p = m.find(target-nums[i]);
if(p!=m.end() && p->second!=i)
{
result.push_back(i);
result.push_back(m[target-nums[i]]);
}
m[nums[i]] = i;
}
if(result[0]>result[1])
swap(result[0],result[1]);
return result;
}