median.h

来自「包括ARM开发环境的使用和一些调试用的源程序」· C头文件 代码 · 共 32 行

#ifndef GUARD_median_h
#define GUARD_median_h

#include <algorithm>
#include <stdexcept>
#include "Vec.h"

using std::domain_error;
using std::sort;

template <class T>
T median(Vec<T> v)
{
#ifdef _MSC_VER
        typedef typename Vec<T>::size_type vec_sz;
#else
	typedef typename Vec<T>::size_type vec_sz;
#endif

	vec_sz size = v.size();
	if (size == 0)
		throw domain_error("median of an empty Vec");

	sort(v.begin(), v.end());

	vec_sz mid = size/2;

	return size % 2 == 0 ? (v[mid] + v[mid-1]) / 2 : v[mid];
}
#endif