HDU5701 中位数计数

中位数计数

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

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） 

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

题目给的8000数据量 n^2勉强能不超吧

中位数 排中间  比他大的 和 比他小的 一样多 

一个点 往两边扩散  左边 大的数量多  右边小的数量多  二则相等则仍是中位数

#include<bits/stdc++.h>
#define lld I64d
using namespace std;
typedef long long ll;
const int N = 8000 + 10;
int a[N],aa[2*N];
ll ans[N];
int main()
{
    int n;
    while(scanf("%d",&n) != EOF)
    {
        memset(ans,0,sizeof(ans));
        for(int i = 0 ; i < n ; i ++) scanf("%d",&a[i]);
        for(int i = 0 ; i < n ; i ++)
        {
            int num = 0;
            memset(aa,0,sizeof(aa));
            for(int j = i - 1 ; j >= 0 ; j --)
            {
                if(a[j] < a[i]) num--;
                else num++;
                if(!num) ans[i] ++;
                aa[num + N] ++;
            }
            num = 0;
            for(int j = i + 1 ; j < n ; j ++)
            {
                if(a[j] < a[i]) num--;
                else num ++;
                if(!num) ans[i] ++;
                ans[i] += aa[-num + N];//一次加上左边所有的相符的数量
            }
        }
        for(int i = 0 ; i < n ; i ++)
        {
            printf("%lld",ans[i] + 1);
            i == n - 1 ? printf("\n") : printf(" ");
        }
    }
    return 0;
}