【LeetCode】1. Two Sum
目录
暴力解法【时间:$O(n^2)$ 空间:$O(1)$】
思路
对于数组中的每一个位置的数字x
,寻找位于该数字后方的所有数字中,是否有能与它匹配的数字target-x
。
代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
n = len(nums)
for i in range(n):
for j in range(i + 1, n):
if nums[i] + nums[j] == target:
return [i, j]
return []
使用哈希表【时间:$O(n)$ 空间:$O(n)$】
思路
从题意出发,可以想到:进行遍历时,使用哈希表记录已经遍历过的数字和数字的索引,对于当前位置的值x
,在哈希表中寻找能与它匹配的数字target-x
。
代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashtable = dict()
for i, num in enumerate(nums):
if target - num in hashtable:
return [hashtable[target - num], i]
hashtable[nums[i]] = i
return []