题目链接
Problem Description
中位数定义为所有值从小到大排序后排在正中间的那个数,如果值有偶数个,通常取最中间的两个数值的平均数作为中位数。
现在有n个数,每个数都是独一无二的,求出每个数在多少个包含其的区间中是中位数。
Input
多组测试数据
第一行一个数n(n≤8000)
第二行n个数,0≤每个数≤109,
Output
N个数,依次表示第i个数在多少包含其的区间中是中位数。
Sample Input
5
1 2 3 4 5
Sample Output
1 2 3 2 1
Source
2016"百度之星" - 初赛(Astar Round2B)
思路:对于每个数来说,它作为中位数有三种情况,第一种是【x,i-1】这段区间的数的中位数是a【i】,第二种是【i+1,y】这段区间的数的中位数是a【i】,第三种是【x,y】这段区间的数的中位数是a【i】,对于第一种和第二种情况我们很好考虑,第二种情况只要记录以下右边比a【i】大的数和比a【i】小的数的个数差为0的时候,那么位置就是满足条件,第一种情况同理,那么怎么处理第三种情况呢,我们可以想一下如果在a【i】右边的数比a【i】大的有2个,在a【i】的左边比a【i】小的数有2个,这样a【i】是不是就可以成为第三种情况的中位数了
,num数组记录的就是对于每个a【i】右边比它大的数和比它小的数的差的个数,由于可能会产生负值,所以取maxn为起点。
#include<bits/stdc++.h>
using namespace std;
const int maxn=8000;
int n,num[maxn<<1],a[maxn<<1];
int main()
{
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;++i) scanf("%d",&a[i]);
for(int i=1;i<=n;++i)
{
memset(num,0,sizeof(num));
int cnt=0,ans=0;
num[maxn]++;
for(int j=i+1;j<=n;++j)
cnt+=a[j]>a[i]?1:-1,num[maxn+cnt]++;
ans+=num[maxn];
cnt=0;
for(int j=i-1;j>=1;--j)
cnt+=a[j]<a[i]?1:-1,ans+=num[maxn+cnt];
printf("%d%s",ans,i==n?"\n":" ");
}
}
}