题目:两数之和
给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
因为比较菜,所以一开始只想到了暴力法,即用两层循环来解。C代码如下:
int* twoSum(int* nums, int numsSize, int target) {
static int a[2] = {0};
for(int i = 0; i < numsSize; i++) {
for(int j = i+1; j < numsSize; j++) {
if(nums[i]+nums[j] == target) {
a[0] = i;
a[1] = j;
return a;
}
}
}
return 0;
}
简单的两层循环,复杂度是,最后用了244ms。
之后参见了官方提供的解答,发现可以用哈希表来做,可以将复杂度降为。通过对大佬代码的学习,改进的C++代码如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> hash;
int n = (int) nums.size();
for(int i = 0; i<n; i++){
auto p = hash.find(target-nums[i]);
if(p!=hash.end()){
return {p->second,i};
}
hash[nums[i]]=i;
}
return vector<int>();
}
};
采用哈希表,可以是在以近似恒定的时间进行快速查找。在迭代的过程中,将元素插入到哈希表中,同时,可以检查表中是否已经存在该元素的对应元素。如果存在,则表明已求出对应解,可将其返回。通过以时间换空间的方法,我们将时间复杂度从降到了
。同时,空间复杂度从
提升到了
。
对于检查数组中的元素是否存在的问题,使用暴力法进行遍历是不高效的。本题中,我们需要找出数组中的每个元素和它对应的索引。而保持数组中的每个元素与其索引相互对应的最好方法就是哈希表,因此,在解决类似的查询问题时,可以考虑使用哈希表。这样,即可大大提升程序的运行速度。