【LeetCode】1. 两数之和(哈希表)

哈希表

题目

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

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例 :

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

哈希表法

利用哈希表，每次计算target - nums[i] 是否在哈希表中，如果哈希表中存在，则找到答案返回。若不存在，则将nums[i]插入哈希表中。
时间复杂度O(N) , 空间复杂度O(N)

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int,int> map;
        for (int i = 0 ; i < nums.size() ; i ++){
            int t = target - nums[i];
            //哈希表找到target - nums[i]，则表明找到答案
            if (map.count(t)) return {map[t] , i};
            map[nums[i]] = i;//将该数插入哈希表
        }
		
		//题目保证有答案，这里不会执行，不过为了防止报错，必须加
        return {};
    }
};