51Nod 1682 中位数计数

个人博客→TanJX的自留地

中位数计数

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

现在有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

题意

中文题

题解

以某一个数向左右发散才能保证这个数包含在这个区间内。向左向右遍历时比a[i]大的+1,比a[i]小的-1,以一个k确定区间和。刚开始我是想直接用一个数组存每个数到i的区间和，但是无法保证每个区间都能算到所以WA了。。。后来换成用数组存每个k出现的次数，左边存，右边计算保证区间和为０的区间个数。以n为对称点来保证区间和为0。可惜当时改代码着急（因为时间要到了）忘记了加上以i为右端点的区间数量。。。又WA了一发。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>

using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 8010;

int main()
{
    int n;
    int a[maxn],sum[maxn],p[2*maxn];
    while(cin>>n)
    {
        memset(a,0,sizeof a);
        memset(sum,0,sizeof sum);
        for(int i = 1; i <= n; i++)
            scanf("%d",&a[i]);
        for(int i = 1; i <= n; i++)
        {
            memset(p,0,sizeof p);
            int k = 0;
            p[n]++;
            for(int j = i-1; j > 0 ; j--)
            {
                if(a[j] < a[i])
                    k--;
                else
                    k++;
                p[k+n]++;　//以n对称
            }
            k = 0;
            sum[i]+=p[n]; 　　//加上以i为右端点的区间和位0的区间数量，之前就是把它着急忘了。。。
            for(int j = i+1; j <= n; j++)
            {
                if(a[j] < a[i])
                    k--;
                else
                    k++;
                sum[i] += p[n-k];　//左区间加右区间区间和为0的区间数量
            }
        }
        for(int i = 1; i <= n; i++)
        {
            if(i == n)
                printf("%d\n",sum[i]);
            else
                printf("%d ",sum[i]);
        }
    }
    return 0;
}