HDU-5701 中位数计数(水题)

中位数计数

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

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

题目分析：

刚开始没头绪，一个个区间求中位数肯定超时。后来看见 https://blog.csdn.net/qq_34374664/article/details/53218124 这篇博客，思路是比这个数大的数字和比这个数小的数字一样多的时候，这个数就是中位数，茅塞顿开。
对要计算的数字，从它向左遍历，在向右遍历，用数组b记录相应区间比它大或小的个数。
有2种情况这个数字可能是中位数：

1.这个数在某个区间中比它大的数字和比它小的数字一样多。
2.这个数在左遍历时，若在某个区间大于它的数字比小于它的数字多x个，在向右遍历时，某个区间小于它的数字比大于它的数字多x个，那么这两个区间合在一起组成的区间中这个数是中位数。显而易见，两个符合1中的情况的左右区间也可以拼起来。

对每个数t进行左遍历时，用b[n+x]来表示大于t的数-小于t的数=x的区间数，若在右遍历时，每找到一个小于t的数-大于t的数=x的区间，都可以合成一个t是中位数的区间。
例如：{5，4，2，3，1}中，对3进行查找，向左遍历。
遍历到2时比3大的数有-1个，所以b[n+(-1)]++;注意，这里的n只是是个偏移量，因为x可能是负数，所以需要偏移量。
遍历到4时，比3大的数有0个（抵消了），所以b[n+0]++。
遍历到5时，比3大的数有1个，所以b[n+1]++。
现在向右遍历，遍历到1时，比3小的数有-1个，跟前面比3大有-1个的可组成合法区间，所以sum+=b[n+(-1)]。
大概就是这样。

代码实现：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int a[8006],b[16008];
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(a,0,sizeof(a));

        for(int i=0;i<n;i++) scanf("%d",&a[i]);
        for(int i=0;i<n;i++)
        {
            memset(b,0,sizeof(b));
            int sum=0,num=0;
            b[n]=1;
            for(int j=i-1;j>=0;j--)
            {
                if(a[i]>a[j]) num--;
                if(a[i]<a[j]) num++;
                b[num+n]++;
            }
            sum+=b[n];
            num=0;
            for(int j=i+1;j<n;j++)
            {
                if(a[i]>a[j]) num++;
                if(a[i]<a[j]) num--;
                sum+=b[n+num];
            }
            printf("%d%c", sum, i == n-1? '\n' : ' ');

        }
    }
}