EZOPS interview question

Leetcode 2-sum problem in Python