Two sum

Problem Statement

Given an array of integers nums and an integer target, return indices of the two numbers such that they add up to target. LeetCode link

Code

Python Code

class Solution:
    def twoSum(self, nums, target: int):

        num_set = {}
        for num_index, num in enumerate(nums):
            if (target-num) in num_set:
                return [num_set[target-num], num_index]
            num_set[num] = num_index