題目:
給定一個整數數組
nums
和一個整數目标值
target
,請你在該數組中找出 和為目标值
target
的那 兩個 整數,并傳回它們的數組下标。
你可以假設每種輸入隻會對應一個答案。但是,數組中同一個元素在答案裡不能重複出現。
你可以按任意順序傳回答案。
示例 1:
輸入:nums = [2,7,11,15], target = 9
輸出:[0,1]
解釋:因為 nums[0] + nums[1] == 9 ,傳回 [0, 1] 。
示例 2:
輸入:nums = [3,2,4], target = 6
輸出:[1,2]
示例 3:
輸入:nums = [3,3], target = 6
輸出:[0,1]
提示:
-
2 <= nums.length <= 104
-
-109 <= nums[i] <= 109
-
-109 <= target <= 109
- 隻會存在一個有效答案
代碼:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> map;
for (int i = 0; i < nums.size(); i++){
auto iter = map.find(target - nums[i]);
if (iter != map.end()){
return {i, iter->second};
}
map[nums[i]] = i;
}
return {};
}
};