P1627 [CQOI2009] 中位数

P1627 [CQOI2009] 中位数

小清新 trick 题。
题意：给你一个长度为 n 的排列，数 b，求中位数为 b 的子段个数。
容易想到，把 > b 的作为 1，< b 的作为 -1，最后统计包含 p 且和为 0 的子段个数。
用 map 维护一下左边不同和的个数，再往右扫一遍就做完了。

#include<bits/stdc++.h>
#define rep(i, a, b) for(int i = (a); i <= (b); ++ i)
#define per(i, a, b) for(int i = (a); i >= (b); -- i)
#define pb emplace_back
#define All(X) X.begin(), X.end()
using namespace std;
using ll = long long;
constexpr int N = 1e5 + 5;

int n, b, p, a[N], s;
ll ans;
map<int, int> mp;

int main() {
	ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
	cin >> n >> b;
	rep(i, 1, n) {
		cin >> a[i];
		if(a[i] == b) p = i;
	}
	mp[0] = 1;
	per(i, p - 1, 1) ++ mp[s += a[i] > b ? 1 : -1];
	s = 0;
	rep(i, p + 1, n) ans += mp[s += a[i] > b ? -1 : 1];
	cout << ans + mp[0];
	return 0;
}