My solution
Professor’s solution