此题数据过小，建议修改。

甚至不需要nth_element，直接用时间复杂度$O(nlog n)$的sort后查询也可通过。