題目:
給定一個(gè)整數(shù)數(shù)組 nums 和一個(gè)目標(biāo)值 target,請(qǐng)你在該數(shù)組中找出和為目標(biāo)值的那 兩個(gè) 整數(shù),并返回他們的數(shù)組下標(biāo)。
你可以假設(shè)每種輸入只會(huì)對(duì)應(yīng)一個(gè)答案。但是,你不能重復(fù)利用這個(gè)數(shù)組中同樣的元素。
示例 1:
給定 nums = [2, 7, 11, 15], target = 9
因?yàn)?nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
題目分析:
暴力的話時(shí)間復(fù)雜度為O(n2),所以使用map來爭取一下時(shí)間,畢竟map的查找速度可是O(logn)的。
(其實(shí)算法題碰到查找的話,都可以用map來做。)
C++代碼如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int, int> m;
vector<int> result(2);
for (int i = 0; i < nums.size(); ++i) {
m[nums[i]] = i;
}
for (int i = 0; i < nums.size(); ++i) {
int value = target - nums[i];
int index = m.count(value);
if (index && m[value] != i) {
result[0] = i;
result[1] = m[value];
break;
}
}
return result;
}
};
