51Nod ~ 1682 ~ 中位数计数 (思维)

思路:ans[i]存储答案,枚举每一个数字a[i],对于a[i],先往左边扫,维护一个变量cnt和数组sum,

如果a[j]>a[i],cnt++,如果a[j]<a[i],cnt--,那么cnt表示的意思为大于a[i]的数字比小于a[i]的数字多多少个(为-表示少多少个)。每次sum[cnt]++,sum[cnt]的意思为(大于a[i]的数字比小于a[i]的数字多cnt个)的地方有几个。

完成以后我们再往右边扫描,注意从i位置开始,cnt=0,维护cnt,每次ans[i] += sum[-cnt]。意思为如果这一边大于a[i]的数字比小于a[i]的数字多cnt个,那么他与左边 (大于a[i]的数字比小于a[i]的数字少cnt个)的区间构成一个a[i]是中位数的区间。

这样n^2的复杂度就OK啦。


#include<bits/stdc++.h>
using namespace std;
const int MAXN = 8e3 + 5;
const int INF = 0x3f3f3f3f;
int n, a[MAXN], sum[MAXN*2], ans[MAXN];
int main()
{
    while (~scanf("%d", &n))
    {
        for (int i = 0; i < n; i++) scanf("%d", &a[i]);
        for (int i = 0; i < n; i++)
        {
            memset(sum, 0, sizeof(sum));
            int cnt = 0;
            for (int j = i; j >= 0; j--)
            {
                if (a[j] > a[i]) cnt++;
                if (a[j] < a[i]) cnt--;
                sum[8000+cnt]++;
            }
            cnt = 0;
            for (int j = i; j < n; j++)
            {
                if (a[j] > a[i]) cnt++;
                if (a[j] < a[i]) cnt--;
                ans[i] += sum[8000-cnt];
            }
        }
        for (int i = 0; i < n; i++)
        {
            printf("%d", ans[i]);
            if (i != n-1) printf(" ");
            else printf("\n");
        }
    }
    return 0;
}
/*
5
1 2 3 4 5
*/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值