每个人都有那么一段安静的岁月!
题目描述
给定一个整型数组,要求返回两个数的下标,使得两数之和等于给定的目标值,要求同一个下标不能使用两次。
数据保证有且仅有一组解。
样例
给定数组 nums = [2, 7, 11, 15],以及目标值 target = 9,
由于 nums[0] + nums[1] = 2 + 7 = 9,
所以 return [0, 1].
算法1
(暴力枚举) O(n2)
暴力枚举方法很简单:两重循环枚举下标 i,j然后判断nums[i]+nums[j] 是否等于 targettarget。
时间复杂度:由于有两重循环,所以复杂度是 O(n2)
C++ 代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target)
{
vector<int> res;
for (int i = 0; i < nums.size(); i++)
{
for (int j = 0; j < i; j++)
{
if (nums[i] + nums[j] == target)
{
res = vector<int>({j, i});
break;
}
}
if (res.size() > 0) break;
}
return res;
}
};
算法2
(哈希表) O(n)
使用C++中的哈希表——unordered_map<int, int> hash.
循环一遍 nums 数组,在每步循环中我们做两件事:
判断 target−nums[i] 是否在哈希表中;
将 nums[i] 插入哈希表中;
解释:由于数据保证有且仅有一组解,假设是 i,j则我们循环到 j 时,nums[i]一定在哈希表中,且有 nums[i]+nums[j]==target, 所以我们一定可以找到解。
时间复杂度:由于只扫描一遍,且哈希表的插入和查询操作的复杂度是 O(1),所以总时间复杂度是 O(n)
C++ 代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target)
{
vector<int> res;
unordered_map<int,int> hash;
for (int i = 0; i < nums.size(); i ++ )
{
int another = target - nums[i];
if (hash.count(another))
{
res = vector<int>({hash[another], i});
break;
}
hash[nums[i]] = i;
}
return res;
}
};