码迷,mamicode.com
首页 > 其他好文 > 详细

Median on Segments (Permutations Edition)

时间:2020-01-11 18:36:07      阅读:74      评论:0      收藏:0      [点我收藏+]

标签:++i   main   problem   --   href   logs   pos   www   htm   

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;
}

Median on Segments (Permutations Edition)

标签:++i   main   problem   --   href   logs   pos   www   htm   

原文地址:https://www.cnblogs.com/CADCADCAD/p/12180394.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!