每天一算：Two Sum

最新推荐文章于 2024-06-14 05:46:10 发布

chongruxie2626

最新推荐文章于 2024-06-14 05:46:10 发布

阅读量90

点赞数

文章标签：数据结构与算法

原文链接：https://my.oschina.net/u/4010368/blog/2353533

版权

leetcode上第1号问题：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

动画演示

Two Sum

代码

// 1. Two Sum
// https://leetcode.com/problems/two-sum/description/
// 时间复杂度：O(n)
// 空间复杂度：O(n)
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {

        unordered_map<int,int> record;
        for(int i = 0 ; i < nums.size() ; i ++){
        
            int complement = target - nums[i];
            if(record.find(complement) != record.end()){
                int res[] = {i, record[complement]};
                return vector<int>(res, res + 2);
            }

            record[nums[i]] = i;
        }
    }
};