题目描述
给定一个整型数组,要求返回两个数的下标,使得两数之和等于给定的目标值,要求同一个下标不能使用两次。
数据保证有且仅有一组解。
样例
给定数组 nums = [2, 7, 11, 15],以及目标值 target = 9,
由于 nums[0] + nums[1] = 2 + 7 = 9,
所以 return [0, 1].
算法1:
(暴力枚举) O(n2)
暴力枚举方法很简单:
(1)两重循环枚举下标 i,j,
(2)然后判断 nums[i]+nums[j] 是否等于 target。
时间复杂度:由于有两重循环,所以复杂度是
O
(
n
2
)
O(n2)
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 数组,在每步循环中我们做两件事:
(1)判断 target−nums[i] 是否在哈希表中;
(2)将 nums[i] 插入哈希表中;
解释:由于数据保证有且仅有一组解,假设是
[
i
,
j
]
(
i
<
j
)
[i,j](i<j)
[i,j](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;
}
};
~~~![执行空间图](https://img-blog.csdnimg.cn/20200223182922351.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMzODUyODYz,size_16,color_FFFFFF,t_70)