hdu4908 序列计数（中位数）

最新推荐文章于 2024-07-21 14:30:00 发布

原创最新推荐文章于 2024-07-21 14:30:00 发布 · 355 阅读

0 ·

CC 4.0 BY-SA版权

－－－Tip－－－专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种算法，用于求解给定数组中以指定数值作为中位数的所有连续子序列的数量。通过统计大于或小于该中位数的元素来实现这一目标。

求以m为中位数的连续子序列的个数。
那么统计m位置往右的大于（小于）m的个数相等位置往左小于（大于）m的个数的个数即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[40005],b[80005];
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        int pos;
        memset(b,0,sizeof(b));
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            if(a[i]==m) pos=i;
        }
        int cnt=0,ans=0;
        b[n]=1;
        for(int i=pos+1;i<n;i++)
        {
            if(a[i]>m) cnt++;
            else cnt--;
            b[n+cnt]++;
        }
        cnt=0;
        for(int i=pos-1;i>=0;i--)
        {
            if(a[i]>m) cnt++;
            else cnt--;
            ans+=b[n-cnt];
        }
        ans+=b[n];
        printf("%d\n",ans);
    }
}