中位数定义为所有值从小到大排序后排在正中间的那个数,如果值有偶数个,通常取最中间的两个数值的平均数作为中位数。
现在有 n个数,每个数都是独一无二的,求出每个数在多少个包含其的区间中是中位数。
现在有 n个数,每个数都是独一无二的,求出每个数在多少个包含其的区间中是中位数。
Input多组测试数据
第一行一个数n(n≤8000)n(n≤8000)
第二行nn个数,0≤0≤每个数≤109≤109,OutputNN个数,依次表示第ii个数在多少包含其的区间中是中位数。Sample Input
5 1 2 3 4 5
Sample Output
1 2 3 2 1
解法:
对于每个数字,
从该数字左边开始扫描,统计个数,遇到比它小的+1,比它大的-1,遇到0,答案+1
从该数字右边开始扫描,遇到比它小的-1,比它大的+1,遇到0,答案+1
然后左边右边的差个数相等的,乘起来,加到答案中。
代码:
#include<iostream>
using namespace std;
#include<cstdio>
#include<cstring>
int a[30000];
int cs1[21000],cs2[21000];
int main(){
int n;
while(scanf("%d",&n)!=EOF){
for(int i=0;i<n;i++) scanf("%d",&a[i]);
for(int i=0;i<n;i++){
int l1=0,l2=0;
memset(cs1,0,sizeof(cs1));
memset(cs2,0,sizeof(cs2));
long long ans=1;
for(int j=i-1;j>=0;j--){
if(a[j]<a[i]) l1++;else l1--;
if(l1==0) ans++;
cs1[l1+n]++;
}
for(int j=i+1;j<n;j++){
if(a[j]>a[i]) l2++;else l2--;
if(l2==0) ans++;
cs2[l2+n]++;
}
for(int j=0;j<=2*n;j++)
ans+=(long long)cs1[j]*cs2[j];
printf("%lld",ans);
if(i!=n-1) putchar(' ');
}
putchar('\n');
}
return 0;
}