求无序数组中的中值

Feliz Da Vida

于 2024-07-17 15:20:32 发布

阅读量49

点赞数 2

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/qq_45512728/article/details/140496302

版权

算法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

template <typename T>
double Median(const std::vector<T>& elems) {
  THROW_CHECK(!elems.empty());

  const size_t mid_idx = elems.size() / 2;

  std::vector<T> ordered_elems = elems;
  
  /* std::nth_element 是 C++ 标准库算法中的一个非常有用的函数，它定义在 <algorithm> 头文件中。
  该函数主要用于对序列中的元素进行部分排序，以便序列中第 n 个位置的元素处于其最终排序位置，同时保证所有小于它的元素都位于其前面，
  所有大于它的元素都位于其后面。但是，它并不保证其他元素的相对顺序。*/

  std::nth_element(ordered_elems.begin(),
                   ordered_elems.begin() + mid_idx,
                   ordered_elems.end());

  if (elems.size() % 2 == 0) {
    const T mid_element1 = ordered_elems[mid_idx];
    const T mid_element2 = *std::max_element(ordered_elems.begin(),
                                             ordered_elems.begin() + mid_idx);
    return 0.5 * mid_element1 + 0.5 * mid_element2;
  } else {
    return ordered_elems[mid_idx];
  }
}