CSP python csp201612_1 100分

试题编号： 201612-1 
试题名称： 中间数 
时间限制： 1.0s 
内存限制： 256.0MB 
问题描述： 
　　在一个整数序列a1, a2, …, an中，如果存在某个数，大于它的整数数量等于小于它的整数数量，则称其为中间数。在一个序列中，可能存在多个下标不相同的中间数，这些中间数的值是相同的。给定一个整数序列，请找出这个整数序列的中间数的值。 
输入格式 
　　输入的第一行包含了一个整数n，表示整数序列中数的个数。 
　　第二行包含n个正整数，依次表示a1, a2, …, an。 
输出格式 
　　如果约定序列的中间数存在，则输出中间数的值，否则输出-1表示不存在中间数。 
样例输入 
6 
2 6 5 6 3 5 
样例输出 
5 
样例说明 
比5小的数有2个，比5大的数也有2个。 
评测用例规模与约定 

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

python代码

n = int(input())
nums = list(map(lambda x:int(x),list(input().split(' '))))
nums.sort()
index_mid = n // 2
mid = nums[index_mid]
count_less = 0
for i in range(0, index_mid):
    if nums[i] < mid:
        count_less += 1

count_larger = 0
for i in range(index_mid + 1, n):
    if nums[i] > mid:
        count_larger += 1

count = 1
if count_less == count_larger:
    print(mid)
    for i in range(n):
        if i == mid:
            count += 1
    print("比%d小的数有%d个，比%d大的数也有%d个。"%(mid,count_less,mid,count_larger))

else:
    print(-1)