Two sum python geeksforgeeks