中位数计数
中位数定义为所有值从小到大排序后排在正中间的那个数,如果值有偶数个,通常取最中间的两个数值的平均数作为中位数。
现在有n个数,每个数都是独一无二的,求出每个数在多少个包含其的区间中是中位数。
Input 第一行一个数n(n<=8000)
第二行n个数,0<=每个数<=10^9 Output N个数,依次表示第i个数在多少包含其的区间中是中位数。 Sample Input
5 1 2 3 4 5Sample Output
1 2 3 2 1
a[i]是区间的中位数 即 区间内 大于a[i]的数的个数A = 小于a[i] 的数的个数B;
即 i 左边的 A-B 的值 = i 右边的 B - A 的值;
cnt 为 x-i 区间内 A - B 的值 (x∈[ 0 , i ));
对于每一个 a[i],在 0 - i 求一个 m 数组;
m[cnt] 表示 [ 0 , i ) 内 A - B = cnt 的区间终点为i的区间起点个数;
初始化 m[0] =1,此区间只包含 a[i] ;
ans[i] 为 i 是中位数 区间的个数
遍历[ i ,n ) 再算一遍 cnt 则此时的区间个数即 m[-1*cnt] ;
(我写的 cnt 为 B - A ,则此时的区间个数为 m[cnt])
最后 ans[i] 求一遍和即可;
#include <iostream>
#include <cstdio>
#include <cstring>
#define ll long long
using namespace std;
ll a[8050];int ans[8050];
int m[16050];
int main(){
int n;
cin>>n;
for(int i=0;i<n;i++)
scanf("%lld",&a[i]);
int cnt;
for(int i=0;i<n;i++){
memset(m,0,sizeof(m));
m[8000]=1;
ans[i]=0;
cnt=0;
for(int j=i-1;j>=0;j--){
if(a[j]>a[i]) cnt++;
else cnt--;
m[cnt+8000]++;
}
cnt=0;
for(int j=i;j<n;j++){
if(a[j]>a[i]) cnt--;
else if(a[j]<a[i]) cnt++;
ans[i]+=m[cnt+8000];
}
}
for(int i=0;i<n;i++){
if(i) printf(" ");
printf("%d",ans[i]);
}
cout<<endl;
return 0;
}