写的时候太匆忙了，使用的是 std::set 和 std::distance。需要注意的是 std::distance 的时间复杂度是线性的，除非迭代器为 random-access iterators，时间复杂度为常数（std::vector 属于这一种情况）。参考文献：https://en.cppreference.com/w/cpp/iterator/distance。所以如果你也用的是 std::set，还是换成 pb_ds 吧。