Pregunta de entrevista de Tencent

I did in python, biggest K elements