Leetcode 1. 两数之和（哈希）

题目

给一个数组，和一个目标值 target，要找到数组中两个数加起来等于 target，并返回两个数下标，注意两个数不能是相同的数。只有一种答案。

分析

当然可以暴力，O(n^2)枚举所有两个数的可能。

O(n) 做法是，利用两个数之间的联系，固定一个数 a，另外一个数也随之固定：target - a。可以用哈希预处理数组，将第二个循环优化掉。

代码思路就是，因为一定存在一对数符合，可以固定第二个数，在当前位置前面找第一个数。

🐖：注意 map 底层红黑树，平均 O(logn)；unordered_map 底层哈希表，平均 O(1)；

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> heap;
        for (int i = 0; i < nums.size(); i ++) {
            int r = target - nums[i];       // 如果很长的表达式后面频繁要用，最好替换一下
            if (heap.count(r)) return {heap[r], i};
            heap[nums[i]] = i;
        }
        return {};  // 防止报错
    }
};