HelloFresh interview question

The task was implemented using pyspark