【CCF CSP-20161201】中间数
题意概述
在一个整数序列
a
1
,
a
2
,
⋯
,
a
n
a_1,a_2,\cdots,a_n
a1,a2,⋯,an中,如果存在某个数,大于它的整数数量等于小于它的整数数量,则称其为中间数。在一个序列中,可能存在多个下标不相同的中间数,这些中间数的值是相同的。
给定一个整数序列,请找出这个整数序列的中间数的值。
输入输出格式
输入的第一行包含了一个整数 n,表示整数序列中数的个数。第二行包含 n 个正整数,依次表示 a 1 , a 2 , ⋯ , a n a_1,a_2,\cdots,a_n a1,a2,⋯,an。
如果约定序列的中间数存在,则输出中间数的值,否则输出-1,表示不存在中间数。
数据规模
1 ≤ n ≤ 1000 , 1 ≤ a i ≤ 1000 1\le n\le1000,1\le a_i\le1000 1≤n≤1000,1≤ai≤1000
算法设计
显然,如果存在这样的中间数,则该中间数一定是序列的中位数。可以先将读入的序列排序。这样可以很轻松地找到中位数,那么如何判断这个序列中小于中位数和大于中位数的数的个数是否相等呢?由于排序后的序列是一个有序序列,我们可以使用泛型算法库中的 equal_range 算法,它可以返回两个迭代器,第一个指向序列中第一个大于等于中位数的数的位置,第二个指向序列中第一个大于中位数的数的位置,那么通过这两个迭代器就可以得出序列中小于中位数和大于中位数的数的个数了。
C++代码
#include <bits/stdc++.h>
using namespace std;
using gg = long long;
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
gg ni;
cin >> ni;
vector<gg> ai(ni);
for (gg& i : ai) {
cin >> i;
}
sort(ai.begin(), ai.end());
gg t = ai[ai.size() / 2];
auto p = equal_range(ai.begin(), ai.end(), t);
cout << (p.first - ai.begin() == ai.end() - p.second ? t : -1);
return 0;
}