Virtasant Interview Question

Solve this problem from leet code: [leet code problem]