LeetCode上面的难度为easy的一道题,作为进入Leetcode的第一道题,做下来比较顺利,不过跑的结果不好,原因在于算法的复杂度高,下面给出做这道题遇到的问题
问题描述
解法
根据题目很容易得出第一种解法,直接暴力求解即可
遍历数组,找出符合要求的两个数,并返回
以下是C++代码
#include <vector>
using namespace std;
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
for (int i = 0; i < nums.size(); ++i) {
for (int j = i + 1; j < nums.size(); ++j) {
if (nums[i] + nums[j] == target) {
std::vector<int> v;
v.push_back(i);
v.push_back(j);
return v;
}
}
}
}
};
这是我做的第一种方法,函数要返回两个数,因此创建了一个 vector 返回
假如在java中,则可以使用
return new int[] { i, j };
但在C++中不能这么使用,原因在于C++中并不存在Garbage Collection,内存的分配和释放必须靠自己
因此返回数组的时候显得十分麻烦
该算法的复杂度是 O(N^2)
另一种解法是使用哈希表,在寻找特定值以及下标的时候可以将复杂度降到O(1),遍历的时候O(N),总复杂度O(N)
#include <vector>
#include <map>
using namespace std;
class Solution {
public:
static vector<int> twoSum(vector<int>& nums, int target) {
/* for (int i = 0; i < nums.size(); ++i) {
for (int j = i + 1; j < nums.size(); ++j) {
if (nums[i] + nums[j] == target) {
std::vector<int> v;
v.push_back(i);
v.push_back(j);
return v;
}
}
}*/
map<int, int> mymap;
for (int i = 0; i < nums.size(); ++i) {
mymap.insert( std::pair<int, int>(nums[i], i) );
}
for (int i = 0; i < nums.size(); ++i) {
int complement = target - nums[i];
auto it = mymap.find(complement);
if (it != mymap.end() && it->second != i) {
std::vector<int> v;
v.push_back(i);
v.push_back(it->second);
return v;
}
}
/* std::vector<int> v;
v.push_back(0);
v.push_back(0);
return v;
*/
}
};
这里采用的是C++ STL中的map
结果显示这种做法效率更高