中位数计数
Time Limit: 12000/6000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1734 Accepted Submission(s): 631
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
向一个方向进行统计,每出现一个比当前数小的数,用num–记录,出现比他大的数则
num++记录,若num为0,则证明发现一个区间(一个比它大,一个小,刚好位于中间数),若每边的查询后发现有不对称的(比如3 5 6 4 10 8 7 以4向左查询,则num为1,我们用vist记录vist[num]++,再对右边进行处理时则有num为1,2,3,分别标记vist[1]++,vist[2]++,vist[3]++,这样在右边统计区间时,由于一边一大一小刚好形成一区间,要多加上vist[-num])
遍历每个数,分别向左和向右统计其区间数量,时间复杂度为O(N^2),8000*8000=6.4*1e7,时间足够
#include<iostream>
#include<algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
int main()
{
int n;
int a[8010];
int vist[2*8010];
long long ans[8010];
while(scanf("%d",&n)!=EOF)
{
for(int i=0; i<n; i++)
scanf("%d",&a[i]);
//int num=0;
memset(ans,0,sizeof(ans));
for(int i=0; i<n; i++)
{
int num=0;
memset(vist,0,sizeof(vist));
for(int j=i-1; j>=0; j--)
{
if(a[j]<a[i])num--;
else num++;
if(num==0) ans[i]++;
vist[num+8010]++;
}
num=0;
for(int j=i+1; j<n; j++)
{
if(a[j]<a[i])num--;
else num++;
if(num==0) ans[i]++;
ans[i]+=vist[-num+8010];
}
}
for(int i=0; i<n-1; i++)
{
printf("%I64d ",ans[i]+1);
}
printf("%I64d\n",ans[n-1]+1);
}
return 0;
}