hdu 5701 中位数计数（重构数组）

中位数计数

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

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
  
  
  
  


  
  

思路：网络赛的时候没做出来，惭愧..TUT

对一个数进行重构数组，注意必须保证i是中位数，分为三种情况（注意此题不存在相同的数，即不存在偶数情况的数组）

1：i与其左边的数构成一个新数组，那么初始num=0，如果左边的数比i大，那么num+1，反之-1，每一次num重新变成了0，都意味着大于i与小于i的个数相同，即此时i就是中位数，那么结果ans[i]++ 

2：与1一样。不过是往右边比。

3：仍然要以i为中位数，不过数组是由左边两边组成的，用aa数组记录i往左找时各个num的出现次数。 

在往右边找时，aa[-num]就表示了左右两边比i大和比i小的数目相同。  为了不让aa数组下标变成负数，我们加一个极值N，不影响结果。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define N 8080
using namespace std;
int a[N],ans[N];
int aa[2*N];
int main()
{
   int n,num;
   while(~scanf("%d",&n))
   {
       memset(ans,0,sizeof(ans));

       for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
       for(int i=0;i<n;i++)
       {
           memset(aa,0,sizeof(aa));
           num=0;
           for(int j=i-1;j>=0;j--)
           {
               if(a[j]<a[i]) num--;
               else num++;
               if(!num) ans[i]++;
               aa[N+num]++;
           }
           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[N-num];
           }
       }
       for(int i=0;i<n-1;i++)
           printf("%d ",ans[i]+1);
           printf("%d\n",ans[n-1]+1);
   }
   return 0;
}