Pregunta de entrevista de ByteDance

Coding exercise, list and dictionaries in Python.