LeetCode：1 Two Sum

最新推荐文章于 2024-07-14 11:23:58 发布

笑笔狂生

最新推荐文章于 2024-07-14 11:23:58 发布

阅读量301

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/ls291187101/article/details/50741988

版权

LeetCode 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

1 Two Sum

题目大意就是说给了一个数组和一个目标值，找到数组中相加等于目标值的两个数，返回下标，并且限定了只有一组解。
例子：Given nums = [2, 7, 11, 15], target = 9,return [0, 1].

开始想到的就是暴力解，两重循环套一起，检查到相加等于target就return，代码这样：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result;
        for (int i = 0; i < nums.size(); i++)
        {
            for (int j = i + 1; j < nums.size(); j++)
            {
                if (nums[i] + nums[j] == target)
                {
                    result.push_back(i);
                    result.push_back(j);
                    return result;
                }
            }
        }
    }
};

然后直接AC了。。。卧槽这是O（N^2）的算法啊= =

之后看看有没有更好的算法，在网上看到一个大神用map写的一个O（N）的算法，就是数组的一遍遍历，当map中没有该数的时候，将此数和数的下标插入map，然后检测（target - num）在map中是否存在，如果存在的话直接返回。注意判断两个下标不能相等，这样写出来的代码如下：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector <int> result;
        map <int,int> m;
        for (int i = 0; i < nums.size(); i++)
        {
            if (!m.count(nums[i]))
            {
                m.insert(pair<int, int>(nums[i], i));
            }
            if (m.count(target - nums[i]))
            {
                int j = m[target - nums[i]];
                if (j < i)
                {
                    result.push_back(j);
                    result.push_back(i);
                    return result;
                }
            }
        }
    }
};