LeetCode https://leetcode.cn/problems/two-sum/
题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
思路
将数组转换为map, key为数组中的元素, value为这个元素在数组中的下标
遍历map, 检查target 与 key的差是否在map中
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
| class Solution { public: vector<int> twoSum(vector<int>& nums, int target) { std:unordered_map <int, int> map; int size = (int)nums.size(); for (int i = 0; i < size; ++i) { auto iter = map.find(target - nums[i]); if (iter != map.end()) { return {iter->second, i}; } map[nums[i]] = i; } return {}; } };
|