std :: multiset并找到中间元素

时间:2012-08-27 10:54:26

标签: c++ stl

我有std::multiset如果我从std::multiset::begin()迭代到std::multiset::end()我将获得已排序的元素。除了从std::multiset迭代到std::multiset::begin()

之外,我如何获得此std::multiset::begin() + size() / 2中的中间元素

1 个答案:

答案 0 :(得分:8)

以下是获取std :: multiset中值的解决方案:

template<class T>
double GetMedian(const std::multiset<T>& data)
{
    if (data.empty())
        throw std::length_error("Cannot calculate median value for empty dataset");

    const size_t n = data.size();
    double median = 0;

    auto iter = data.cbegin();
    std::advance(iter, n / 2);

    // Middle or average of two middle values
    if (n % 2 == 0)
    {
        const auto iter2 = iter--;
        median = double(*iter + *iter2) / 2;    // data[n/2 - 1] AND data[n/2]
    }
    else
    {
        median = *iter;
    }

    return median;
}

如果您需要中位数绝对偏差(mdev),例如:

template<class T>
double GetMedianAbsoluteDeviation(const std::multiset<T>& data)
{
    const double fMedian = GetMedian(data);
    std::multiset<double> diff;
    for (const auto& x : data)
    {
        diff.insert(std::abs(fMedian - x));
    }
    return GetMedian(diff);
}