LeetCode - Two Sum 完整代码(C++)
C++暴力法
#include <iostream>
#include <vector>
using namespace std;
vector<int> twoSum(vector<int>& nums, int target) {
int n = nums.size();
for (int i = 0; i < n - 1; ++i) {
for (int j = i + 1; j < n; ++j) {
if (nums[i] + nums[j] == target) {
return {i, j};
}
}
}
return {};
}
int main()
{
int num[5] = { 1,6,7,8,9 };
vector<int> nums(num, num + 5);
int target = 7;
vector<int> result;
result = twoSum(nums, target);
cout << result[0] << "," << result[1] << endl;
return 0;
}
时间复杂度:O(N^2).其中 N 是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。
空间复杂度:O(1)。
\
注意到方法一的时间复杂度较高的原因是寻找 target - x 的时间复杂度过高。因此,我们需要一种更优秀的方法,能够快速寻找数组中是否存在目标元素。如果存在,我们需要找出它的索引。
使用哈希表,可以将寻找 target - x 的时间复杂度降低到从 O(N) 降低到 O(1)
C++哈希表法
#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> hashtable;
for (int i = 0; i < nums.size(); ++i) {
auto it = hashtable.find(target - nums[i]);
if (it != hashtable.end()) {
return { it->second, i };
}
hashtable[nums[i]] = i;
}
return {};
}
int main()
{
int num[5] = { 1,6,7,8,9 };
vector<int> nums(num, num + 5);
int target = 7;
vector<int> result;
result = twoSum(nums, target);
cout << result[0] << "," << result[1] << endl;
return 0;
}
\
复杂度分析
时间复杂度:O(N),其中 N 是数组中的元素数量。对于每一个元素 x,我们可以 O(1) 地寻找 target - x。
空间复杂度:O(N),其中 N 是数组中的元素数量。主要为哈希表的开销。
本题知识点
c++ map与unordered_map区别
需要引入不同的头文件
map: #include < map >
unordered_map: #include < unordered_map >
unordered_map执行效率要比map高
map:优点:有序性
it->first会得到key,
it->second会得到value。