hdu 5701 中位数计数

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=5701

中位数计数

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

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
  
  

 

Source

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

 

Recommend

wange2014   |   We have carefully selected several similar problems for you:   6032  6031  6030  6029  6028 

 

解题思路：开始时想错了，想成是排序之后再判断了，以为没有顺序呢。。。后来才反应过来，要在原序列区间找。。。zz

对于每一个数x，设一个计数器cnt：

1.向其左边遍历，如果遍历到的数字比x小cnt就加1，比x大cnt就减1，这样对于cnt=0的时候就是x是中位数的时候

2.向右遍历，和上面情况相反，如果遍历到的数字比x大cnt就加1，比x小cnt就减1

3.在向右遍历时，对于和向左遍历时cnt相等（比x小的数和比x大的数相等）的情况，也是x在中间的情况

注意：

对于不同cnt出现的次数，我们用一个数组cc[]来记录，为避免cnt为负，初始都加上一个值。。。

单独x时也是一种情况。。。

代码：

#include <cstdio>
#include <cstring>
#define  M 8000

int cc[M * 2 + 5];
int a[M + 5];

int main()
{
	int i, j, n;
	while (~scanf("%d", &n))
	{
		for (i = 1; i <= n; i++)
			scanf("%d", &a[i]);
		for (i = 1; i <= n; i++)
		{
			memset(cc, 0, sizeof(cc));
			int cnt = 0, ans = 0;
			cc[M]++;
			for (j = i-1; j >=1; j--)
			{
				cnt += a[j] < a[i] ? 1 : -1;
				cc[M + cnt]++;
			}
			cnt = 0;
			ans += cc[M];
			for (j = i + 1; j <= n; j++)
			{
				cnt += a[j] > a[i] ? 1 : -1;
				ans += cc[M + cnt];
			}
			printf("%d%s", ans, i == n ? "\n" : " ");
		}
	}
	return 0;
}