Python Hackerearth Solution