51nod 1682-中位数计数（前缀和）

1682 中位数计数

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

中位数定义为所有值从小到大排序后排在正中间的那个数，如果值有偶数个，通常取最中间的两个数值的平均数作为中位数。

现在有n个数，每个数都是独一无二的，求出每个数在多少个包含其的区间中是中位数。

Input

第一行一个数n(n<=8000)
第二行n个数，0<=每个数<=10^9

Output

N个数，依次表示第i个数在多少包含其的区间中是中位数。

Input示例

5
1 2 3 4 5

Output示例

1 2 3 2 1

Joe  (题目提供者)

C++的运行时限为：1000 ms ，空间限制为：131072 KB  示例及语言说明请按这里

 允许其他 AC 的用户查看此代码，分享代码才能查看别人的代码并有机会获得勋章

题解：一开始用了很烦的n^2的方法，A了这道题，后来发现利用前缀和便能很容易搞掉这道题，我们可以定义一个sum数组，用来统计前缀和，统计谁的前缀和呢，我们假设枚举到当前第i个数字，令比这个数大的数为1，比它小的数为-1，相等则为0，然后统计一发前缀和，前边的可以O(n)求出，后边可以O(1)查询，总复杂度为O(n^2)...

#include<map>   
#include<stack>          
#include<queue>          
#include<vector>  
#include<string>
#include<math.h>          
#include<stdio.h>          
#include<iostream>          
#include<string.h>          
#include<stdlib.h>  
#include<algorithm> 
#include<functional>  
using namespace std;          
typedef long long  ll;         
#define inf  1000000000     
#define mod 1000000007           
#define maxn  8050
#define lowbit(x) (x&-x)          
#define eps 1e-9
int ans[maxn],sum[maxn],a[maxn],num[maxn*3];
int main(void)
{
	int n,i,j,tmp;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(i=1;i<=n;i++)
	{
		tmp=sum[0]=0;
		memset(num,0,sizeof(num));
		num[maxn]++;
		for(j=1;j<=n;j++)
		{
			if(a[j]<a[i]) sum[j]=sum[j-1]-1;
			else if(a[j]==a[i]) sum[j]=sum[j-1];
			else sum[j]=sum[j-1]+1;
			if(j<i) ++num[sum[j]+maxn];
			else tmp+=num[sum[j]+maxn];
		}
		ans[i]=tmp;
	}
	for(i=1;i<=n;i++)
		printf("%d ",ans[i]);
	printf("\n");
	return 0;
}