CCF CSP 201612-1 中间数

CCF计算机职业资格认证考试题解系列文章为meelo原创，请务必以链接形式注明本文地址

CCF CSP 201612-1 中间数

问题描述

　　在一个整数序列a₁, a₂, …, a_n中，如果存在某个数，大于它的整数数量等于小于它的整数数量，则称其为中间数。在一个序列中，可能存在多个下标不相同的中间数，这些中间数的值是相同的。
　　给定一个整数序列，请找出这个整数序列的中间数的值。

输入格式

　　输入的第一行包含了一个整数n，表示整数序列中数的个数。
　　第二行包含n个正整数，依次表示a₁, a₂, …, a_n。

输出格式

　　如果约定序列的中间数存在，则输出中间数的值，否则输出-1表示不存在中间数。

样例输入

6
2 6 5 6 3 5

样例输出

5

样例说明

　　比5小的数有2个，比5大的数也有2个。

样例输入

4
3 4 6 7

样例输出

-1

样例说明

　　在序列中的4个数都不满足中间数的定义。

样例输入

5
3 4 6 6 7

样例输出

-1

样例说明

　　在序列中的5个数都不满足中间数的定义。

评测用例规模与约定

　　对于所有评测用例，1 ≤ n ≤ 1000，1 ≤ a_i ≤ 1000。

解析

首先排序，中间数如果存在那么一定会是a[N/2]。

然后再统计比a[N/2]小的数和比它大的数。

复杂度受排序制约为O(nlogn)

代码

C++

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

int main() {
    int N;
    cin >> N;
    vector<int> arr(N);
    for(int n=0; n<N; n++) {
        cin >> arr[n];
    }
    sort(arr.begin(), arr.end());
    int num = arr[N/2];
    int cntSmall = 0;
    int cntBig = 0;
    for(int n=0; n<N; n++) {
        if(arr[n] < num) cntSmall++;
        if(arr[n] > num) cntBig++;
    }
    if(cntSmall==cntBig) cout << num;
    else cout << -1;
}