最近看了遍C++ primer开始刷leetcode,巩固C++知识,正好复习数据结构和算法。
第一题: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.
Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
我的代码:
class Solution
{
public:
vector<int> twoSum(vector<int>& nums,int target)
{
vector<int> ivec;
for(vector<int>::size_type i=0;i!=nums.size();++i)
{
for(vector<int>::size_type j=i+1;j!=nums.size();++j)
{
if(nums[i]+nums[j]==target)
{
ivec.push_back(i);
ivec.push_back(j);
}
}
}
return ivec;
}
};
本来看了问题的tags显示array和hash table,说明得用到散列表,但是没想到怎么用,果然自己的代码和答案的solution1类似,用的brute force algorithm。
此时我的代码时间复杂度为O(n2),空间复杂度为O(n).
因为每对应一个i都遍历j为size-i-1次,时间复杂度为O(n),所以总体时间复杂度为O(n2),此时的时间复杂度很大。
而散列表(hash表)可以快速的查找一个具体值,为什么可以快速的查找一个值呢?散列表这种思想对应C++语言的map容器(键值对),现实生活中字典这个例子可以很好的说明这种数据结构,单词即为键,单词的解释即为值。正如查字典一样,查找单词“algorithm”,先查找字母a,再查找l……每个单词在字典的查找时间为O(1),散列表是用于实现固定时间内完成的操作。
class Solution
{
public:
vector<int> twoSum(vector<int>& nums,int target)
{
vector<int> ivec;
multimap<int,int> imap;
typedef multimap<int, int> ::iterator iter_map;
int complement=0;
for(vector<int>::size_type i=0;i!=nums.size();++i)
{
imap.insert(map<int,int>::value_type(nums[i],i));
}
for(iter_map iter=imap.begin();iter!=imap.end();++iter)
{
cout<<iter->first<<" ";
}
cout<<endl;
for(iter_map iter=imap.begin();iter!=imap.end();++iter)
{
cout<<iter->second<<" ";
}
cout<<endl;
for(iter_map iter=imap.begin();iter!=imap.end();++iter)
{
complement=target-iter->first;
iter_map beg=imap.lower_bound(complement);
iter_map end=imap.upper_bound(complement);
while(beg!=end)
{
if(beg!=iter)
{
ivec.push_back(iter->second);
ivec.push_back(beg->second);
}
beg++;
}
imap.erase(iter->first);
}
return ivec;
}
};
本来是用的map容器满足他测试的例子,后来提交时候例子变为了{0,4,3,0},也就是会出现冲突-键相同的值(下标)有两个。此时更改为multimap容器了。而且multimap容器中是按照顺序排列的,即将数组转换为multimap变为了{0,0,3,4},使用multimap自带的函数进行操作即可。每次将匹配过得数据删除,以防重复使用,不然会出现结果为{0,3,3,0},而正确结果为{0,3}。