Median on Segments (Permutations Edition)

原创

caoanda 2022-11-03 15:18:46 博主文章分类：CF ©著作权

©著作权归作者所有：来自51CTO博客作者caoanda的原创作品，请联系作者获取转载授权，否则将追究法律责任

E1. Median on Segments (Permutations Edition)

参考：CF1005E1 Median on Segments (Permutations Edition) 思维

中位数为m的条件为，在那一段中，小于 m 的数的个数为 x 个，大于 m 的数有 y 个，要满足条件x==y||x==y-1。
因为不可能每一次都去统计有多少个大于的多少个小于的，所以我们要预处理一下，借此来降低复杂度
map<int,int> ma; int cnt=0; for(int i=pos;i>=1;--i) { if(p[i]<m) cnt--; else if(p[i]>m) cnt++; ma[cnt]++; }

代码：

// Created by CAD on 2020/1/11.
#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int maxn=2e5+5;
int p[maxn];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m;cin>>n>>m;
    int pos;
    for(int i=1;i<=n;++i)
    {
        cin>>p[i];
        if(p[i]==m) pos=i;
    }
    map<int,int> ma;
    int cnt=0;
    for(int i=pos;i>=1;--i)
    {
        if(p[i]<m) cnt--;
        else if(p[i]>m) cnt++;
        ma[cnt]++;
    }
    ll ans=0;  cnt=0;
    for(int i=pos;i<=n;++i)
    {
        if(p[i]<m) cnt++;
        else if(p[i]>m) cnt--;
        ans+=ma[cnt]+ma[1+cnt];
    }
    cout<<ans<<endl;
    return 0;
}