Polaris Wireless interview question

Write a program in Python