Domanda di colloquio di Apple

Write a function that passes the tests below.