方法一:暴力。
#include <iostream>
#include <vector>
using namespace std;
void printVector(vector<int> &v) //引用传递 减少值传递方式的拷贝时间
{
//定义了一个名为it的变量,它的数据类型是由vector<int>定义的iterator类型
for (vector<int>::iterator it = v.begin(); it != v.end(); it++)
cout << *it << "\t"; //*it代表<int>
cout << endl;
}
vector<int> twoSum(vector<int>& nums, int target)
{
/***以下为提交的代码***/
int len = nums.size();//提前记录长度 代码会更快
for (int i = 0; i < len - 1; i++)
{
for (int j = i + 1; j < len; j++)
{
if ((nums[i] + nums[j]) == target)
{
return { i,j };
}
}
}
return { 0,0 };
/***以上为提交的代码***/
}
void test01()
{
vector<int> v;
v.push_back(2);
v.push_back(7);
v.push_back(11);
v.push_back(15);
int target = 9;
vector<int> result=twoSum(v, target);
printVector(result);
}
int main()
{
test01();
system("pause");
return 0;
}
- 时间复杂度:O(N2),其中 N 是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。
- 空间复杂度:O(1)。
方法二:哈希
注意到方法一的时间复杂度较高的原因是寻找 target - x 的时间复杂度过高。因此,我们需要一种更优秀的方法,能够快速寻找数组中是否存在目标元素。如果存在,我们需要找出它的索引。
使用哈希表,可以将寻找 target - x 的时间复杂度降低到从 O(N)降低到 O(1)。
这样我们创建一个哈希表,对于每一个 x,我们首先查询哈希表中是否存在 target - x,然后将 x 插入到哈希表中,即可保证不会让 x 和自己匹配。
value存坐标,key存数据。判断target-x得到的数能不能和当前的key匹配,由于map查找O(1),所以比方法一更快。
注意使用unordered_map,比map开销更小。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> unmap; //元素,下标
vector<int> ret;
for(int i=0;i<nums.size();i++) {
int temp=target-nums[i];
if(unmap.find(temp)==unmap.end()) //未找到
unmap.insert(make_pair(nums[i],i));
else { //找到
ret.push_back(unmap[temp]);
ret.push_back(i);
return ret;
}
}
return ret;
}
};
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ret;
map<int,int> m; //用于存储结果 差,坐标
int t; //存储临时数值
for(int i=0;i<nums.size();i++){
t=target-nums[i];
if(m.find(t)==m.end()){ //未匹配
m.insert(make_pair(nums[i],i)); //放入当前坐标
}
else{ //查找到差值
ret.push_back(m[t]);
ret.push_back(i);
}
}
return ret;
}
};
#include <iostream>
#include <vector>
#include <map>
using namespace std;
void printVector(vector<int>& v) //引用传递 减少值传递方式的拷贝时间
{
//定义了一个名为it的变量,它的数据类型是由vector<int>定义的iterator类型
for (vector<int>::iterator it = v.begin(); it != v.end(); it++)
cout << *it << "\t"; //*it代表<int>
cout << endl;
}
void printMap(map<int, int>& m)
{
for (map<int, int>::iterator it = m.begin(); it != m.end(); it++)
cout << "key: " << (*it).first << "\tvalue: " << (*it).second << endl;
cout << endl;
}
vector<int> twoSum(vector<int>& nums, int target)
{
/***以下为提交的代码***/
int len = nums.size();
map<int, int> m;
map<int, int>::iterator pos = m.end();
for (int i = 0; i < len; ++i)
{
pos = m.find(target - nums[i]);
if (pos != m.end())
return { pos->second,i };
m[nums[i]] = i;
}
return { 0,0 };
/***以上为提交的代码***/
}
void test01()
{
vector<int> v;
v.push_back(2);
v.push_back(7);
v.push_back(11);
v.push_back(15);
int target = 9;
vector<int> result = twoSum(v, target);
printVector(result);
}
int main()
{
test01();
system("pause");
return 0;
}