我正在编写代码,从矢量中删除最小值和最大值,然后取剩余值的平均值。
我有这样的代码:
#include <vector>
#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
double averageWithoutOutliers(vector<int> values) {
double max = *max_element(values.begin(),values.end());
values.erase(std::remove(values.begin(), values.end(), max), values.end());
double min = *min_element(values.begin(),values.end());
values.erase(std::remove(values.begin(), values.end(), min), values.end());
auto n = values.size();
double average = 0.0;
if ( n != 0) {
average = accumulate( values.begin(), values.end(), 0.0) / n;
return average;
}
else
return min;
}
这段代码可以工作,除非有多个最大值。如何检查重复的最大值,或者只删除重复值中的一个?
std::remove()
删除所有匹配指定值的元素。如果您只想删除一个元素,那么使用vector::erase()
代替:
double averageWithoutOutliers(vector<int> values) {
if (values.size() < 2)
return ...; // or throw...
auto iter = max_element(values.begin(), values.end());
values.erase(iter);
iter = min_element(values.begin(), values.end());
double min = *iter;
values.erase(iter);
if (!values.empty())
return accumulate(values.begin(), values.end(), 0.0) / values.size();
return min;
}
另外:
double averageWithoutOutliers(vector<int> values) {
if (values.size() < 2)
return ...; // or throw...
auto iters = minmax_element(values.begin(), values.end());
auto min_iter = iters.first;
double min = *min_iter;
auto max_iter = iters.second;
if (max_iter < min_iter)
std::swap(min_iter, max_iter);
values.erase(max_iter);
values.erase(min_iter);
if (!values.empty())
return accumulate(values.begin(), values.end(), 0.0) / values.size();
return min;
}
我会用稍微不同的方式来处理它:
double average(const std::vector<double>& values) {
if (values.size() < 2)
return 0.0;
auto ends = std::minmax_element(values.begin(), values.end());
double sum = std::accumulate(values.begin(), values.end(),
-(*ends.first + *ends.second));
return sum / (values.size() - 2);
}