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

解题思路

  1. 这个题目大多数人看一眼就能想出思路,即对数组进行两次遍历,然后每次循环挑出一个元素,组成一个元素对。只要判断此元素对是不是等于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()作为条件。这点是在编译原理课程中学到的。

  2. 上述思路显然是相当简单的,提交后也能正确。但是显然它不是最优解法,时间复杂度还可以进行优化。想到这里,我们可以用空间来换取时间,即使用一个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更加快,具体原因打算再写一篇博文,有兴趣的同学可以自行查询相关资料。

结语

总体来说,第一题还是相当简单的,并没有用到什么复杂的算法知识,而且第一题我之前还做过。。。

希望后面能坚持刷题,至少每天一道。以后这种水题也就不写博客了,太浪费时间。

希望此代码能给大家带来收获,谢谢大家!

坚持不懈地努力才能成为大神!

posted @ 2021-02-03 22:45  己平事  阅读(153)  评论(0编辑  收藏  举报