parent
d3c2184af8
commit
acdd16bceb
|
@ -0,0 +1,25 @@
|
|||
def two_sum(nums, target):
|
||||
# Create a dictionary to store the elements and their indices
|
||||
num_dict = {}
|
||||
|
||||
# Traverse the array
|
||||
for i, num in enumerate(nums):
|
||||
# Calculate the complement required to reach the target
|
||||
complement = target - num
|
||||
|
||||
# Check if the complement exists in the dictionary
|
||||
if complement in num_dict:
|
||||
# Return the indices of the two numbers that add up to the target
|
||||
return [num_dict[complement], i]
|
||||
|
||||
# Store the current element and its index in the dictionary
|
||||
num_dict[num] = i
|
||||
|
||||
# If no pair is found, return [-1, -1]
|
||||
return [-1, -1]
|
||||
|
||||
# Test example
|
||||
nums = [2, 7, 11, 15]
|
||||
target = 9
|
||||
result = two_sum(nums, target)
|
||||
print(result) # Output: [0, 1] (Because nums[0] + nums[1] = 2 + 7 = 9)
|
Loading…
Reference in New Issue