1. Two sum
給定一個整數數組 nums 和一個整數 target,返回兩個數字的索引,使它們相加等於 target。
你可以假設每個輸入都有 恰好一個解,並且你不能使用相同的元素兩次。
你可以按任意順序返回答案。
範例
輸入:nums = [2,7,11,15], target = 9
輸出:[0,1]
解釋:因為 nums[0] + nums[1] == 9,所以返回 [0, 1]。
Python
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
num_to_index = {}
for i, num in enumerate(nums):
complement = target - num
if complement in num_to_index:
return [num_to_index[complement], i]
num_to_index[num] = i
return []
17.74MB 56ms
C++
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> numMap;
for (int i = 0; i < nums.size(); i++) {
int complement = target - nums[i];
if (numMap.find(complement) != numMap.end()) {
return {numMap[complement], i};
}
numMap[nums[i]] = i;
}
return {};
}
};
13.99MB 0ms
Javascript
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
var twoSum = function(nums, target) {
let numToIndex = new Map();
for (let i = 0; i < nums.length; i++) {
let complement = target - nums[i];
if (numToIndex.has(complement)) {
return [numToIndex.get(complement), i];
}
numToIndex.set(nums[i], i);
}
return [];
};
54.4MB, 62ms