中位数是所有值从小到大排序后排在正中间的那个数,如果值有偶数个,通常取最中间的两个数值的平均数作为中位数。
现在有n个数,保证n个数中无重复数字,求出每个数在多少个包含其的区间中是中位数。
Input
多组测试数据
第一行一个数n(n<8000)
第二行n个数,0<每个数<=1e9,
Output
N个数,依次表示第i个数在多少包含其的区间中是中位数。
Sample Input
5
1 2 3 4 5
Sample Output
1 2 3 2 1
这道题分三种情况,这个点是这个区间的右端点,左端点,和在中间的情况,前两种都好说,直接向前向后遍历 即可,设置一个变量cnt,每次遇到比自己大的数加一,比自己小的数减一,当cnt为0时这个区间满足,答案加一即可,但是如果这个点是第三种情况的话,就需要用到一个小技巧,向前遍历时用一个数组记录下出现的所有情况(即cnt的大小)的数量,向后遍历时如果出现-cnt,那么就加上这个情况的数量即可.这个小技巧还蛮重要的,至少我已经连续碰到两次了.
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=8000+100;
int arr[N],ans[N];
int vis[20000];
int main()
{
int n;
while(cin>>n)
{
memset(ans,0,sizeof(ans));
for(int i=0;i<n;i++)
cin>>arr[i];
for(int i=0;i<n;i++)
{
memset(vis,0,sizeof(vis));
int cnt=0;
for(int j=i-1;j>=0;j--)
{
if(arr[j]<arr[i])
cnt--;
else
cnt++;
if(cnt==0)
ans[i]++;
vis[10000+cnt]++;//防止出现负数
}
cnt=0;
for(int j=i+1;j<n;j++)
{
if(arr[j]<arr[i])
cnt--;
else
cnt++;
if(cnt==0)
ans[i]++;
ans[i]+=vis[10000-cnt];
}
}
for(int i=0;i<n-1;i++)
cout<<ans[i]+1<<' ';
cout<<ans[n-1]+1<<endl;
}
return 0;
}