math.h

Go to the documentation of this file.

#pragma once

#include <exception>

namespace tc {

template <typename V>
typename V::value_type median(V v) {
  if (v.size() == 0) {
    throw std::out_of_range("median not defined for empty containers");
  }

  auto n = v.size();
  std::nth_element(v.begin(), v.begin() + n / 2, v.end());
  if (n % 2 == 1) {
    return v.at(n / 2);
  }
  auto rightElement = v.at(n / 2);
  std::nth_element(v.begin(), v.begin() + n / 2 - 1, v.end());
  return (v.at(n / 2 - 1) + rightElement) / 2;
}

} // namespace tc