Leetcode-初级算法-两数之和

题目

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

示例：

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

自己的思路

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

#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的键值对中，同时检查表中是否已经存在当前元素所对应的目标元素。

#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;
}

复杂度分析：

• 时间复杂度：O(n)
  我们只遍历了包含有n个元素的列表一次。在表中进行的每次查找只花费 O(1) 的时间。
• 空间复杂度：O(n)
  所需的额外空间取决于哈希表中存储的元素数量，该表最多需要存储 n 个元素。

貌似使用hash_map效率会更高