LeetCode刷题之路-1. 两数之和
LeetCode刷题之路-1. 两数之和
前言
最近要开始刷题了,打算每天要刷一两道算法题,对于一些难的算法知识点要及时搞懂;然后按时参加LeetCode的周赛和双周赛。立此博客为证,作为监督。
题目介绍
给定一个整数数组 nums 和一个整数目标值 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 <= 103
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
解题思路
-
这个题目大多数人看一眼就能想出思路,即对数组进行两次遍历,然后每次循环挑出一个元素,组成一个元素对。只要判断此元素对是不是等于target即可。
该思路的代码如下:
vector<int> twoSum(vector<int>& nums, int target) { int length = nums.size(); // 将nums.size()从循环中提取出来设置为变量,防止多次调用size()函数 for(int i = 0; i < length; i++){ for(int j = i + 1; j < length; j++){ if(nums[i] + nums[j] == target) return {i, j}; } } return {}; }
减少运行时间的小窍门:nums.size()单独设置成一个length变量作为循环结束条件,不要直接使用nums.size()作为条件。这点是在编译原理课程中学到的。
-
上述思路显然是相当简单的,提交后也能正确。但是显然它不是最优解法,时间复杂度还可以进行优化。想到这里,我们可以用空间来换取时间,即使用一个map结构,将每个元素作为key,元素位置作为value存入map中,这样可以将两层循环变为一层循环,从而降低时间复杂度。
该思路的代码如下:
vector<int> twoSum(vector<int>& nums, int target) { unordered_map<int, int> m; // 不用map,而用unordered_map int length = nums.size(); for(int i = 0; i < length; i++){ auto it = m.find(target - nums[i]); if(it != m.end()){ return {it -> second, i}; } m[nums[i]] = i; } return {}; }
在这里为什么要使用STL库中的unordered_map而不用map呢,是因为unordered_map更加快,具体原因打算再写一篇博文,有兴趣的同学可以自行查询相关资料。
结语
总体来说,第一题还是相当简单的,并没有用到什么复杂的算法知识,而且第一题我之前还做过。。。
希望后面能坚持刷题,至少每天一道。以后这种水题也就不写博客了,太浪费时间。
希望此代码能给大家带来收获,谢谢大家!
坚持不懈地努力才能成为大神!