Sollicitatievraag bij Tencent

I did in python, biggest K elements