Hacker Rank restaurant challenge - contest 1