给定一个整数数组 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]
#####可以使用哈希表存储数组中的每个数字和索引,当我们碰到某个数num的时候,利用哈希表快速查找的特性,判断target - num是否存在于哈希表中。
如果存在,返回索引对。
如果不存在,num作为潜在的target - ?存储到哈希表中,直到它的搭档来找到它。
Java解法:
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i< nums.length; i++) {
if(map.containsKey(target - nums[i])) {
return new int[] {map.get(target-nums[i]),i};
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
}
Python解法:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
m = {} # 哈希表
for i, num in enumerate(nums):
if target - num in m:
# 如果当前数nums[i]的搭档存在,直接返回索引对
return [i, m[target - num]]
m[num] = i # 否则将当前数作为一个潜在目标存储到哈希表中
return [-1, -1] # 不存在满足条件的数对返回-1索引