有没有办法将 nth_element 与数据副本一起做？

Question

我希望从 C++ 中的浮点数组中计算中值：

float Median( FloatArray const * constFloatArray )
{
    FloatArray    scratch = FloatArray( *constFloatArray );
    int64_t const size    = scratch.GetWidth() * scratch.GetHeight();
    int64_t const mid     = size / 2;

    std::nth_element( scratch.begin(), scratch.begin() + mid, scratch.end() );

    return scratch[ mid ];
}

FloatArray 包含一个常规的 C++ 浮点数数组。

我正在使用std::nth_element，但想知道是否有像nth_element 这样的设施可以处理const 数据？现在，我正在制作一个副本，然后在扔掉副本之前做nth_element。如果const 数据没有类似nth_element 的内容，是否有更有效的方法使用复制步骤来计​​算信息，从而避免潜在的额外 O(n) 循环？也许性能影响可以忽略不计？我的数组大小可能在 20 亿左右。

Answer 1

我不确定它是否会更有效，但您可以使用std::partial_sort_copy 节省一半的复制。我们可以使用std::partial_sort_copy 仅将一半数据复制到一个新数组中，然后它会将其排序到该数组中。然后，您需要做的就是为奇数个元素获取最后一个元素，或者为偶数个元素获取最后两个元素的平均值。使用看起来像的向量

int main() 
{
    std::vector<int> v{5, 6, 4, 3, 2, 6, 7, 9, 3, 10};
    std::vector<int> r(v.size() / 2 + 1);
    std::partial_sort_copy(v.begin(), v.end(), r.begin(), r.end());
    if (r.size() % 2)
        std::cout << r.back();
    else
        std::cout << (r[r.size() - 1] + r[r.size() - 2]) / 2.0;
}