中位数计数
Time Limit: 12000/6000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1368 Accepted Submission(s): 513
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
题解:
我们统计每个数字,对于比它大的记一个1,小的记一个-1,然后只要考虑左右相加为0的有多少。
由于不是我的思路,我想了好久,决定用大白话解释一波:
对于一个数来说,它把数列分成左右两个区间,我们分开考虑。
假设对于左区间,我们想知道每个子区间有多少个大于给定数的,多少个小于给定数。这样我们再找一个区间使得两个区间大于的数的个数=小于数的个数,这样给定数就一定是中位数了。
代码:
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include <iostream>
using namespace std;
const int N=1e6;
int a[N],cnt[N];
int n;
int main()
{
while(~scanf("%d",&n))
{
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
for(int i=1; i<=n; i++)
{
for(int j=1; j<=2*n; j++)
{
cnt[j]=0;
}
cnt[n]++;
int x=0;
for(int j=1;j<i;j++)
{
if(a[i-j]<a[i])
x--;
else
x++;
cnt[n+x]++;
}
int ans=cnt[n];
x=0;
for(int j=i+1;j<=n;j++)
{
if(a[j]<a[i])
x--;
else
x++;
ans+=cnt[n-x];
}
printf("%d%c",ans,i==n?'\n':' ');
}
}
}