Function selectPivots

test/test_CORE/test_topk.cpp:13–19

随机选择 pivots

// 随机选择 pivots
std::vector<int> selectPivots(const std::vector<int>& arr, int numPivots) {
    std::vector<int> pivots;
    // std::sample(arr.begin(), arr.end(), std::back_inserter(pivots), numPivots, rng);
    pivots.insert(pivots.end(), arr.end()-numPivots, arr.end());
    std::sort(pivots.begin(), pivots.end()); // 按顺序排列 pivots
    return pivots;
}

// 分区函数：根据 pivots 将数组分成多个区块
std::vector<std::vector<int>> partition(const std::vector<int>& arr, const std::vector<int>& pivots) {