McKinsey & Company interview question

coding questions based on data science