add faster solution of 1

This commit is contained in:
sangge-rockpi 2024-01-19 21:33:45 +08:00
parent 43c2705a63
commit 53077439e6

View File

@ -4,6 +4,7 @@
class Solution: class Solution:
def twoSum(self, nums: list[int], target: int) -> list[int]: def twoSum(self, nums: list[int], target: int) -> list[int]:
"""
answer = [] answer = []
for i in range(len(nums)): for i in range(len(nums)):
another = target - nums[i] another = target - nums[i]
@ -25,6 +26,15 @@ class Solution:
return answer return answer
return answer return answer
"""
num_map = {}
for i, num in enumerate(nums):
complement = target - num
# reduce compare times
if complement in num_map:
return [num_map[complement], i]
num_map[num] = i
return []
test = [3, 2, 4] test = [3, 2, 4]