Tencent Interview Question

I did in python, biggest K elements