HDU 5701：中位数计数

中位数计数

Time Limit: 12000/6000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1026    Accepted Submission(s): 393

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
  
  

 

 

http://acm.hdu.edu.cn/showproblem.php?pid=5701

2016"百度之星" - 初赛（Astar Round2B）

通过题目可得知，a[i]为中位数所在的区间一定是奇区间，
1、枚举每一个i，找出a[i]，即以a[i]为中位数的区间个数
2、a[i]包括以下三种情况
①: i与其左边的数构成一个新数组，初始num=0，如果左边的数比i大，那么num+1，反之-1，每一次num变成0，都意味着大于i与小于i的个数相同，即此时i就是中位数，ans[i]++
②: i与其右边的数构成一个新数组，初始num=0，如果左边的数比i大，那么num+1，反之-1，每一次num变成0，都意味着大于i与小于i的个数相同，即此时i就是中位数，ans[i]++
③: 数组由i的左边和右边共同组成，用另一个数组sum[]记录i往左找时各个num的出现次数，在往右边找时，k[-num]就表示了左右两边比i大和比i小的数目之和相同，即此时i就是中位数，ans[i]++

复杂度：n²

#include<stdio.h>
#include<string.h>
int a[8007], ans[8007];
int sum[16015];
int main(void)
{
	int i, j, n, k;
	while(scanf("%d", &n)!=EOF)
	{
		memset(ans, 0, sizeof(ans));
		for(i=1;i<=n;i++)
			scanf("%d", &a[i]);
		for(i=1;i<=n;i++)
		{
			memset(sum, 0, sizeof(sum));
			k = 0;
			for(j=i-1;j>=1;j--)
			{
				if(a[j]<a[i])	k--;
				else	k++;
				if(k==0)	ans[i]++;
				sum[n+k]++;
			}
			k = 0;
			for(j=i+1;j<=n;j++)  
			{
				if(a[j]<a[i])	k--;
				else	k++;
				if(k==0)	ans[i]++;
				ans[i] += sum[n-k];
			}
		}
		printf("%d", ans[1]+1);
		for(i=2;i<=n;i++)
			printf(" %d",ans[i]+1);
		printf("\n");
	}
	return 0;
}