Update top_k.cpp