Hot100-Day21-T1两数只和

4 阅读1分钟

Day21[26/3/21]T1两数之和

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 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 <= 104
  • -109 <= nums[i] <= 109
  • -109 <= target <= 109
  • 只会存在一个有效答案

进阶:你可以想出一个时间复杂度小于 O(n2) 的算法吗?

解题思路

很简单,你从头到尾遍历每一个元素,检查这个元素的前面有无大小为 target-nums[i] 的元素即可。

然后速度优化在于这个检查,你可以创建一个哈希表用于存储该元素之前的所有元素,因为你只想知道一个具体的元素的存在与否与下标,所以,该元素作为 key 值并用记录其下标作为 value。

(因为哈希表的查找速度是 O(1))

Code

#include <iostream>
#include <unordered_map>
#include <vector>

using namespace std;

class Solution
{
public:
    vector<int> twoSum(vector<int> &nums, int target)
    {
        unordered_map<int, int> mp;
        for (int i = 0; i < nums.size(); i++)
        {
            // 如果互补元素存在
            if (mp.count(target - nums[i]))
            {
                return {mp[target - nums[i]], i};
            }
            else
            {
                // 否则就存入元素
                mp[nums[i]] = i;
            }
        }

        return {-1, -1};
    }
};

auto main() -> int
{
    vector<int> nums{2, 7, 11, 15};
    int target = 9;

    Solution sol;

    cout << "[";
    for (auto index : sol.twoSum(nums, target))
    {
        cout << index << ',';
    }
    cout << "]" << endl;
}