HYSBZ - 1303 中位数图 伪前缀和，区间问题

题意：

给定 1-n 的排列，找到以 b 为中位数的区间（奇数长度）个数；

思路：

这个区间要包含 b，可能是b及其左边的一段区间，或者b及其右边的一段区间，或者左右两边的区间组合；

开始没有看见 1-n 的排列组合，以为是任意的数，那样不容易搞，后来发现1-n这个排列以后就很简单了，

这是这个区间满足b是中位数这个条件的话，那大于他的数的个数等于小于他的数到个数；

左右两边分别计数， On 的复杂度；

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<set>
#include<queue>
#include<stack>
#include<map>
#define PI acos(-1.0)
#define in freopen("in.txt", "r", stdin)
#define out freopen("out.txt", "w", stdout)

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1e5 + 7, maxd = 670000 + 7, mod = 1e9 + 7;
const int INF = 0x7f7f7f7f;

int n, b, id;
int a[maxn];
int l_[maxn], r_[maxn];
map<int, int> mp;

int main() {
    ios::sync_with_stdio(0);
    cin >> n >> b;
    for(int i = 1; i <= n; ++i) {
        cin >> a[i];
        if(a[i] == b) id = i;
    }
    int cnt = 0;
    for(int i = id-1; i > 0; --i) {
        if(a[i] > b) l_[i] = (++cnt);
        if(a[i] < b) l_[i] = (--cnt);
    }
    cnt = 0;
    for(int i = id+1; i <= n; ++i) {
        if(a[i] > b) r_[i] = (++cnt);
        if(a[i] < b) r_[i] = (--cnt);
    }
    int ans = 1;
    for(int i = id-1; i > 0; --i) {
        if(l_[i] == 0) ans ++;
        mp[l_[i]]++;
    }
    for(int i = id+1; i <= n; ++i) {
        if(r_[i] == 0) ans++;
        if(mp.count(-r_[i])) ans += mp[-r_[i]];
    }
    cout << ans << endl;

    return 0;
}