Domanda di colloquio di Tencent

I did in python, biggest K elements