中位数定义为所有值从小到大排序后排在正中间的那个数,如果值有偶数个,通常取最中间的两个数值的平均数作为中位数。现在有n个数,每个数都是独一无二的,求出每个数在多少个包含其的区间中是中位数。
输入
第一行一个数n(n<=8000)
第二行n个数,0<=每个数<=10^9
输出
N个数,依次表示第i个数在多少包含其的区间中是中位数。
输入样例
5
1 2 3 4 5
输出样例
1 2 3 2 1
对每个数进行操作,从每个数往前计算前缀和,比这个数大就加一,比这个数小就减一
因为前缀和可能为零,就操作数组时加了个常数10000
详情见代码
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <vector>
typedef long long ll;
using namespace std;
ll n,a[10000],num[20000],ans[10000],sum;
int main()
{
scanf("%lld",&n);
for(ll i=0;i<n;i++)
scanf("%lld",&a[i]);
memset(ans,0,sizeof(ans));
for(ll i=0;i<n;i++)
{
memset(num,0,sizeof(num));
sum=0;
for(ll j=i;j>=0;j--)//往前计算a[i]前缀和
{
if(a[j]<a[i]) sum+=-1;
else if(a[j]>a[i]) sum+=1;
num[10000+sum]++;
}
sum=0;
for(ll j=i;j<n;j++)
{
if(a[j]<a[i]) sum+=1;
else if(a[j]>a[i])sum+=-1;
ans[i]+=num[10000+sum];
}
}
for(ll i=0;i<n;i++)
printf("%lld ",ans[i]);
return 0;
}