这道题目难度级别:easy。
题意是给定一个整形数taget和一个vector<int>arr,这个target是arr中两个数求和的结果,要求返回arr中这两个数的索引号。
我的解题思路是:(talk is cheap,直接show code吧)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
//
vector<int> result;
//在剩余的里面找
for (int i = 0; i < nums.size(); i++)
{
//已知i位置的数了,desireNum是渴望的另一个数
int desireNum = target - nums[i];
for (int j = i + 1; j < nums.size(); j++)
{
if (nums[j] == desireNum)
{
result.push_back(i);
result.push_back(j);
break;
}
}
}
return result;
}
};
思路很清晰简单。于是,我高兴的提交代码,通过。
但是,,,
我只击败了9.80%的submissions.......
下面的代码是运行时间只有6ms左右的:
vector<int> twoSum(vector<int>& nums, int target)
{
unordered_map<int, int> map;
vector<int> result;
for (int i = 0; i < nums.size(); i++)
{
unordered_map<int, int>::iterator got = map.find(nums[i]);
if (got != map.end()) //find如果没找到就会返回end,如果找到了就不是end
{
result.push_back(got->second);//2
result.push_back(i);//3
return result;
}
else
{
//没有找到的话..
int complement = target - nums[i];
//pair是一种模板类型,可以存储两个值
pair<int, int> pr = make_pair(complement, i);
map.insert(pr);
}
}
return result;
}
相比于上面我自己写的代码,人家的代码为啥可以在那么短的时间内跑出来??
首先,我的算法复杂度是O(n^2)的,而人家的是O(n),另一层循环查找被unordered_map做快速查询代替了,
简单比较map和unordered_map:
1.
unordered_map的运行效率高于map,需要做快速查找删除并且不担心内存量的时候可以使用unordered_map,
如果很在意内存的占用的话可以用map.
2.
map的内部是二叉平衡树(红黑树),而unordered_map内部是哈希表。
总结:学习STL只是第一步,要会用才是真正的算掌握。