哈希表 | LeetCode 1 两数之和

66 阅读1分钟

题目描述

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target  的那 两个 整数,并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

本题思路

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        std::unordered_map<int, int> map;
        for (int i = 0; i < nums.size(); i++) {
            auto item = map.find(target - nums[i]);
            if (item != map.end())
                return {i, item->second};
            map.insert(pair<int, int>(nums[i], i));
        }
        return {};
    }
};

这题利用map 可以存放两个值 map不允许容器中有重复key值元素/multimap允许容器中有重复key值元素

unordered_map 底层实现是利用哈希表效率更高一些

为什么key是nums的数据元素,而value是下标。因为要利用key来检索,对比key要比对比value容易

那么思路就是将遍历的数据元素与下标依次放入unordered_map,再放入的同时比对map中是否有对应元素相加为target

auto it = m.find(target - nums[i]); //这里it被自动推断为map<int, int>::iterator类型