E1. Median on Segments (Permutations Edition)[如何判断无序中位数]

原创于 2018-07-10 14:31:56 发布 · 445 阅读

1 ·

CC 4.0 BY-SA版权

思维同时被 2 个专栏收录

98 篇文章

订阅专栏

模拟

13 篇文章

订阅专栏

本文探讨了E1.MedianonSegments(PermutationsEdition)题目，即在一个长度为n的全排列中找到包含特定数值m的区间，这些区间中m作为中位数的数量。文章介绍了一种有效的算法实现思路，并提供了完整的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

E1. Median on Segments (Permutations Edition)

题意:一个长为n的打乱的全排列,给定m,问包含m的区间中,有多少个区间的中位数是m

思路:假设m的位置为pos,先处理出[pos+1~n]中到达每个点的,比m大和比m小的差值. 假设一个区间可以,那么有大于m的数-小于m的数=1 or 0 那么. 再从[pos-1,1]处理一遍,就处理出所有的答案

#include<bits/stdc++.h>
#define PI acos(-1.0)
#define pb push_back
#define F first
#define S second
using namespace std;
typedef long long ll;
const int N=2e5+5;
const int MOD=1e9+7;
int a[N],sum1[N],sum2[N];
map<int,ll> mp;
int main(void){
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);

    int n,m,pos;
    cin >>n>>m;
    for(int i=1;i<=n;i++){
        cin >> a[i];
        if(a[i]==m) pos=i;
    }
    int now=0;
    mp[now]++;
    for(int i=pos+1;i<=n;i++){
        if(a[i]>m)  now++;
        else    now--;
        mp[now]++;
    }
    now=0;
    ll ans=mp[0]+mp[1];
    for(int i=pos-1;i>=1;--i){
        if(a[i]<m)  now++;
        else    now--;
        ans+=mp[now]+mp[now+1];
    }
    cout << ans << endl;

    return 0;
}