Leetcode-初级算法-两数之和

最新推荐文章于 2022-06-28 18:08:05 发布

Martin.Y

最新推荐文章于 2022-06-28 18:08:05 发布

阅读量188

点赞数

分类专栏：算法文章标签： LeetCode C++

本文链接：https://blog.csdn.net/sinat_28338727/article/details/82732889

版权

算法专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文首发于我的个人Blog阿西BUG，欢迎大家批评指正

题目

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例：

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

自己的思路

遍历每个元素 x，并查找是否存在一个值与 target−x 相等的目标元素。

耗时: 188ms

#include <stdio.h>
#include <iostream>
#include <vector>

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

复杂度分析：
- 时间复杂度：O(n²)
对于每个元素，我们试图通过遍历数组的其余部分来寻找它所对应的目标元素，这将耗费 O(n) 的时间。因此时间复杂度为 O(n²)。
- 空间复杂度：O(1)

网上优秀思路

通过牺牲空间换取速度的方式。
将vector的元素添加到map的键值对中，同时检查表中是否已经存在当前元素所对应的目标元素。

耗时: 8ms

#include <stdio.h>
#include <iostream>
#include <vector>
#include <map>
using namespace std;

vector<int> twoSum(vector<int>& nums, int target) {
    vector<int> vecResult;
    map<int, int> mapResult;
    for (int i = 0; i < nums.size(); i++)
    {
        int complement = target - nums[i];
        auto it = mapResult.find(complement);
        if (it != mapResult.end())
        {
            vecResult.push_back(i);
            vecResult.push_back(it->second);
        }
        mapResult[nums[i]] = i;
    }
    return vecResult;
}