Median on Segments (Permutations Edition)

Posted 2021-03-16 cadcadcad

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>
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]++;
    }
    int 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;
}