前言
第一道做懂的数据结构!
一、题目描述
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9 输出:[0,1] 解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6 输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6 输出:[0,1]
提示:
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
- 只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n2)
的算法吗?
二、设计步骤
法一
读完题目,脑子里最先蹦出来的就是冒泡式的双层循环遍历,检查每个组合然后返回符合条件的下标。这个思路在算法设计中常称为暴力枚举法,不再赘述,代码如下:
class Solution
{
public:
vector<int> twoSum(vector<int>& nums, int target)
{
int n = nums.size();
for (int i = 0; i < n; ++i)
{
for (int j = i + 1; j < n; ++j)
{
if (nums[i] + nums[j] == target)
{
return {i, j};
}
}
}
return {};
}
};
这个方法可以解决这个问题,但缺点就是两层循环时间复杂度为O(n^2),效率很低,试想如果符合要求的组合位于数组最后面的位置,那前面的几次循环就全是无用功了!
法二
本题的核心问题在于以数组内符合要求元素组合的查找,查找依据是两元素之和为一定值,因此查找值为 target-x 的目标元素就是影响复杂度最主要的因素。
我们在学习线性表的多项式存储问题时得出过一个结论:同一问题可以采用不同的存储方式。对于这个问题,让我们考虑一个哈希表(HashTable)的解决方案。
哈希表是一种基于数组的数据结构。我们在访问数组的元素时往往通过数组下标来访问,数据下标只体现元素的存放位置,与具体的值无关,这也导致如果我们需要在数组中查找信息就需要访问数组的每个元素对存放的值进行比对;而哈希表则通过键-值对(Key-Value)的方式存储数据,可以通俗理解为我们在存储数据时把和这个数据的值有关的特征作为它的下标,就像字典根据拼音的首字母缩写为顺序存放数据一样。
因此我们可以创建一个哈希表,将原数组的值作为哈希表的键,将数组下标作为键对应的值,这样我们每次查找目标元素时只需要在哈希表中查找对应的键。
我们利用C++标准库中用于存储键值对的关联容器unordered_map<int, int>
实现哈希表。通过 hashtable.find(target - nums[i])
在哈希表中查找键为 target - nums[i]
的元素。如果找到了,it
将指向该元素的位置(迭代器)。如果没有找到,则 it
等于 hashtable.end()
,表示未找到。
实现代码:
class Solution{
public:
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 {};
}
};
总结
哈希表法之所以获得比暴力法更好的性能,根本在于:对每个元素在查找其目标元素时哈希表的查找方法find()的时间复杂度为O(1),加上外面的循环整个程序复杂度为O(n);而暴力枚举法对每个元素通过一层循环来查找,加上外层循环复杂度为O(n^2)。但二者的空间复杂度分别为O(1)和O(n),哈希表为了提高查找速度,在存储数据时产生了更多的空间花销,因此空间复杂度更高。这种通过增加内存空间的使用来加快算法的执行速度或者降低复杂度的优化策略被称为空间换时间。
上一次做这道题还是七个月前,那时候对数据结构和算法一窍不通,现在回首颇有感触。
“有人相爱,有人夜里开车看海,有人leetcode第一题都做不出来。”
---摘自此题力扣题解高赞评论