暴力比较 o( n2 ) // 752ms
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
for (int i=0; i< nums.size(); i++)
{
for (int j=i+1; j< nums.size(); j++)
{
if (nums[i] + nums[j] == target)
{
res.push_back(i);
res.push_back(j);
return res;
}
}
}
}
};
使用map o(nlgn) // 28ms
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
map<int, int> m;
for (int i=0; i< nums.size(); i++)
{
int sub = target - nums[i];
if (m.count(sub))
{
res.push_back(i);
res.push_back(m[sub]);
return res;
}
else
m[nums[i]] = i;
}
}
};
使用hash table o(n) // 18ms
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> m;
vector<int> res;
for (int i=0; i< nums.size(); i++)
{
int sub = target - nums[i];
if (m.count(sub))
{
res.push_back(i);
res.push_back(m[sub]);
return res;
}else
m[nums[i]] = i;
}
}
};
hash 法只击败了55%的人,应该可以再做常数上的优化。
从这道题学到了stl 中unordered_map 底层是hash 实现的,map底层是平衡二叉树实现的,unordered_map要比map更快一些,但是实现需要额外的内存。