LeetCode之Two Sum

最新推荐文章于 2020-10-03 14:07:25 发布

小鱼Doris

最新推荐文章于 2020-10-03 14:07:25 发布

阅读量259

点赞数

CC 4.0 BY-SA版权

分类专栏： # LeetCode

本文链接：https://blog.youkuaiyun.com/yulijuanxmu/article/details/76766922

LeetCode 专栏收录该内容

6 篇文章

订阅专栏

本文解析了一道经典算法题“两数之和”，介绍了两种解决方案：暴力解法和利用unordered_map的高效解法。通过对这两种方法的对比，帮助读者理解如何优化算法效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:
Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

暴力解决方法：

class Solution {
public:
    vector twoSum(vector& nums, int target) {
        bool flag = 0;
        vector result;
        int temp_size = nums.size()-1;// if don't use this variable,just write it in for(;i


注意两点：
1.如果将写成for(int i=0; i<sums.size()-1;i++),leetcode会报compile error，这时需要一个变量来存sums.size()-1;
2.发现第一个for循环里面如果只遍历到数组倒数第二个数，但是用一个单独变量来存储sums.size()-1，总的运行时间会比循环sums.size()次稍慢。

精简一点版本：
使用unordered_map
class Solution {
public:
    vector twoSum(vector& nums, int target) {
        vector result;
        unordered_map findTheRest;
        for (int i = 0; i < nums.size(); i++)
        {
            int theRest = target - nums[i];
            if (findTheRest.find(theRest) != findTheRest.end())
            {
                result.push_back(findTheRest[theRest]);
                result.push_back(i);
                break;
            }
            findTheRest[nums[i]] = i;
        }
        return result;
    }
};

原理一样，运行更快的版本：

class Solution {
public:
    vector twoSum(vector& nums, int target) {
        unordered_map findTheRest;
        for (int i = 0; i < nums.size(); i++)
        {
            int theRest = target - nums[i];
            if (findTheRest.find(theRest) != findTheRest.end())
            {
                return{findTheRest[theRest],i};
            }
            findTheRest[nums[i]] = i;
        }
        return {};
    }
};