对向量的整数元素进行排序

时间:2012-09-02 16:00:47

标签: c++ algorithm sorting vector

我想计算存储在矢量中的元素的中位数

#include <vector>
#include <string>
#include <algorithm>
#include <iostream>
using namespace std;

 ....
 ....
 ....

vector<int> trmVa;
int size;
int median;
int trimVal;

trmVa.push_back(trimVal);
size = trmVa.size();
sort(trmVa.begin(), trmVa.end()); //I am having troubles here!!!!

if(size % 2) {
  median = (trmVa[size/2 - 1] + trmVa[size/2]) /2;
  printf("Module %d \n\n \t Median = %d\n", mod, median); 
}else {
   median = trimVa[size/2];
   printf("Module %d \n\n \t Median = %d\n", mod, median);
}

错误:运算符 - 未定义为vector&gt; :: iterator algo.h:722。我很感激帮助。

1 个答案:

答案 0 :(得分:3)

您可以使用std::nth_element更有效地解决此问题。这只会对矢量进行部分排序,并具有线性复杂度。这是一个奇数大小的矢量的例子:

size_t midIndex = trmVa.size()/2;
nth_element(trmVa.begin(), tmrVa.begin() + midIndex, trmVa.end());

中值是

trmVa[midIndex];

您可以轻松扩展它以覆盖偶数大小的矢量。