Coding Problem

Company: amazon_14feb

Difficulty: medium