两数之和

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

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

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

【思路】这个题目使用哈希表可以将算法优化到O(n), 这里面我们只需要遍历一遍哈希表,有一个优化的思路,就是哈希表边创建边查找。其中键值key为数组nums的元素,而value为数组nums的元素的索引。因此当我们遍历到nums[i]时,我们就需要去哈希表中查找target-nums[i],从而得到其索引,因此res将i和hashmap[target-nums[i]]放入数组中!

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> hashmap;
        vector<int> res;
        for(int i = 0;i < nums.size();i++){
            if(hashmap.count(target-nums[i]) > 0){   
                // 比使用hashmap.find(target-nums[i] != hashmap.end())快一点
                res.push_back(hashmap[target-nums[i]]);
                res.push_back(i);
                return res;
            }
            hashmap[nums[i]] = i;
        }
        return res;
    }
};
原文地址:https://www.cnblogs.com/zhudingtop/p/11496096.html