思路
维护一个哈希表,记录每个元素出现的位置。
判断访问到当前元素nums[i]时,是否已经出现了target-nums[i]。
如果出现了,则说明哈希表中存在这个元素,那么直接返回两个元素对应的下标即可。
如果没有,则记录当前元素对应的下标,然后继续访问。
Python版本代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
table = {}
for i in range(len(nums)):
if target - nums[i] in table:
return [table[target - nums[i]],i]
else:
table[nums[i]] = i
Java版本代码
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> hashtable = new HashMap<Integer,Integer>();
for (int i = 0;i<nums.length;i++){
if (hashtable.containsKey(target-nums[i])){
return new int[]{hashtable.get(target-nums[i]),i};
}
hashtable.put(nums[i],i);
}
return new int[]{0};
}
}