Sollicitatievraag bij Apple

Write a function that passes the tests below.