51nod 1682 中位数计数

最新推荐文章于 2020-04-19 10:11:48 发布

沈尽欢

最新推荐文章于 2020-04-19 10:11:48 发布

阅读量156

点赞数 1

分类专栏：前缀和 ---------其他----------

本文链接：https://blog.csdn.net/qq_41758381/article/details/86487120

版权

前缀和同时被 2 个专栏收录

2 篇文章 0 订阅

订阅专栏

---------其他----------

2 篇文章 0 订阅

订阅专栏

该博客探讨了如何在独特的数值集合中计算每个数作为中位数出现的次数。通过从前缀和的角度出发，当数值大于当前数时计数加一，小于则减一，以解决这个问题。代码实现中为了避免前缀和为零，引入了一个常数10000。

摘要由CSDN通过智能技术生成

中位数定义为所有值从小到大排序后排在正中间的那个数，如果值有偶数个，通常取最中间的两个数值的平均数作为中位数。现在有n个数，每个数都是独一无二的，求出每个数在多少个包含其的区间中是中位数。

输入

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

输出

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

输入样例

5
1 2 3 4 5

输出样例

1 2 3 2 1

对每个数进行操作，从每个数往前计算前缀和，比这个数大就加一，比这个数小就减一

因为前缀和可能为零，就操作数组时加了个常数10000

详情见代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <vector>
typedef long long ll;
using namespace std;
ll n,a[10000],num[20000],ans[10000],sum;
int main()
{
    scanf("%lld",&n);
    for(ll i=0;i<n;i++)
        scanf("%lld",&a[i]);
    memset(ans,0,sizeof(ans));

    for(ll i=0;i<n;i++)
    {
        memset(num,0,sizeof(num));
        sum=0;
        for(ll j=i;j>=0;j--)//往前计算a[i]前缀和
        {
            if(a[j]<a[i]) sum+=-1;
            else if(a[j]>a[i]) sum+=1;
            num[10000+sum]++;
        }
        
        sum=0;
        for(ll j=i;j<n;j++)
        {
            if(a[j]<a[i]) sum+=1;
            else if(a[j]>a[i])sum+=-1;
            ans[i]+=num[10000+sum];
        }
    }
    for(ll i=0;i<n;i++)
        printf("%lld ",ans[i]);


    return 0;
}