HDU5701 中位数计数

中位数定义为所有值从小到大排序后排在正中间的那个数，如果值有偶数个，通常取最中间的两个数值的平均数作为中位数。

现在有n个数，每个数都是独一无二的，求出每个数在多少个包含其的区间中是中位数。

 收起

输入

第一行一个数n(n<=8000)
第二行n个数，0<=每个数<=10^9

输出

N个数，依次表示第i个数在多少包含其的区间中是中位数。

输入样例

5
1 2 3 4 5

输出样例

1 2 3 2 1

这道题分三种情况,这个点是这个区间的右端点,左端点,和在中间的情况,前两种都好说,直接向前向后遍历 即可,设置一个变量cnt,每次遇到比自己大的数加一,比自己小的数减一,当cnt为0时这个区间满足,答案加一即可,但是如果这个点是第三种情况的话,就需要用到一个小技巧,向前遍历时用一个数组记录下出现的所有情况(即cnt的大小)的数量,向后遍历时如果出现-cnt,那么就加上这个情况的数量即可.

AC代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<stdlib.h>
#include<queue>
#include<map>
#include<iomanip>
#include<math.h>
#include<sstream>
using namespace std;
typedef long long ll;
typedef double ld;
using namespace std;
const int N=8000+100;
int a[N],ans[N];
int vis[20000]; 
int n,cnt;
int i,j,k;
int main()
{
    while(cin>>n)
    {
        memset(ans,0,sizeof(ans));
        for(i=0; i<n; i++)
            cin>>a[i];
        for(i=0; i<n; i++)
        {
            memset(vis,0,sizeof(vis));
            cnt=0;
            for(j=i-1; j>=0; j--)
            {
                if(a[j]<a[i])
                    cnt--;
                else
                    cnt++;
                if(cnt==0)
                    ans[i]++;
                vis[10000+cnt]++;//防止出现负数
            }
            cnt=0;
            for(int j=i+1; j<n; j++)
            {
                if(a[j]<a[i])
                    cnt--;
                else
                    cnt++;
                if(cnt==0)
                    ans[i]++;
                ans[i]+=vis[10000-cnt];
            }
        }
        for(int i=0; i<n-1; i++)
            cout<<ans[i]+1<<' ';
        cout<<ans[n-1]+1<<endl;
    }
    return 0;
}