2

我希望从 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在扔掉之前做。如果没有类似nth_elementfor constdata 的东西，是否有更有效的方法使用复制步骤来计​​算信息，从而避免潜在的额外 O(n) 循环？也许性能影响可以忽略不计？我的数组大小可能在 20 亿左右。

4

1 回答 1

3

我不确定它是否会更有效，但您可以使用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;
}
于 2019-09-06T20:10:25.990 回答