Minimum Change — Interview Query python problem