Codeforces1005 E1 - Median on Segments (Permutations Edition)

题目：点击打开链接
题意：给出一个1-n的排列，找出中位数为m的连续序列个数。

分析：首先，可以知道答案序列一定存在m，所以可以以m为中心，先求前面（或者后面）大于m与小于等于m的数的个数之差，保存下来，然后记录后面（或者前面）大于m与小于等于m的数的个数之差，求出对应的使得序列中位数为m的数量。细节见代码。

代码：

#pragma comment(linker, "/STACK:102400000,102400000")///手动扩栈
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cstdio>
#include<bitset>
#include<vector>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<deque>
#include<list>
#include<set>
#include<map>
using namespace std;
#define debug test
#define mst(ss,b) memset((ss),(b),sizeof(ss))
#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
#define inf 0x3f3f3f3f
#define eps 1e-10
#define MOD 1000000007
#define PI acos(-1.0)
const int N = 1e6+10;

ll gcd(ll p,ll q){return q==0?p:gcd(q,p%q);}
int to[4][2]={{-1,0},{1,0},{0,-1},{0,1}};

int n,m,a[N],pos;
ll ans;
map<int,int> ma;

int main() {
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++) {
        cin>>a[i];
        if(a[i]==m) pos=i;
    }
    int cnt=0;
    for(int i=pos;i>=1;i--) {
        if(a[i]>m) cnt++;
        else cnt--;
        ma[cnt]++;
    }
    cnt=0;
    ans += ma[0],ans += ma[-1];
    for(int i=pos+1;i<=n;i++) {
        if(a[i]>m) cnt++;
        else cnt--;
        ans += ma[-cnt];
        ans += ma[-cnt-1];
    }
    cout<<ans<<endl;
    return 0;
}