每天一算：Two Sum

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9  
所以返回 [0, 1]

思路

使用查找表来解决该问题。

设置一个map容器record用来记录元素的值与索引

遍历数组nums

• 每次遍历时使用临时变量complement用来保存目标值与当前值的差值

• 在此次遍历中查找record，查看是否有与complement一致的值，如果查找成功则返回查找值的索引值与当前变量的值i

• 如果未找到，则在record保存该元素与索引值i

动画演示

代码

 1// 1. Two Sum
 2// https://leetcode.com/problems/two-sum/description/
 3// 时间复杂度：O(n)
 4// 空间复杂度：O(n)
 5class Solution {
 6public:
 7    vector<int> twoSum(vector<int>& nums, int target) {
 8
 9        unordered_map<int,int> record;
10        for(int i = 0 ; i < nums.size() ; i ++){
11
12            int complement = target - nums[i];
13            if(record.find(complement) != record.end()){
14                int res[] = {i, record[complement]};
15                return vector<int>(res, res + 2);
16            }
17
18            record[nums[i]] = i;
19        }
20    }
21};

执行结果