Macquarie Group Interview Question

- code two sum in python