解题思路
思维题
我们要找的中位数,只和b的相对大小有关系,原本这个数是多大,没什么必要存储。
那么我们可以去吧这个输入的a数组简化一下,比b小的记为负一,比b大的记为正一。b自己记作0。
这样问题就来到了整个区间中,存在几个连续区间和为0?而且一定要包括b在内,也就是从b左边衍生和右边衍生出来,衍生长度可以为0。
把左区间出现过的值次数累计,直接把和为0的,加上b自己一个数,(所以这里方案数为0的增加1)先累加进总方案数。
再去右区间找与他当前对应相反的数再前面出现的次数,累加进答案即可。没有桶的索引是负数,所以如果负数要用另外一个数组去记。
#include <bits/stdc++.h> #pragma GCC optimize(2) #pragma GCC optimize(3) using namespace std; #define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0) typedef long long ll; const ll MOD = 1e9 + 7; inline ll read() { ll s = 0, w = 1; char ch = getchar(); while (ch < 48 || ch > 57) { if (ch == '-') w = -1; ch = getchar(); } while (ch >= 48 && ch <= 57) s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar(); return s * w; } inline void write(ll x) { if (!x) { putchar('0'); return; } char F[200]; ll tmp = x > 0 ? x : -x; if (x < 0)putchar('-'); int cnt = 0; while (tmp > 0) { F[cnt++] = tmp % 10 + '0'; tmp /= 10; } while (cnt > 0)putchar(F[--cnt]); } inline ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; } ll qpow(ll a, ll b) { ll ans = 1; while (b) { if (b & 1) ans *= a; b >>= 1; a *= a; } return ans; } ll qpow(ll a, ll b, ll mod) { ll ans = 1; while (b) { if (b & 1)(ans *= a) %= mod; b >>= 1; (a *= a) %= mod; }return ans % mod; } inline int lowbit(int x) { return x & (-x); } const int N = 1e5 + 7; int a[N], z[N], f[N]; int st; int main() { int n = read(), b = read(); for (int i = 1; i <= n; ++i) { a[i] = read(); if (a[i] > b) a[i] = 1; else if (a[i] < b) a[i] = -1; else st = i, a[i] = 0; } int ans = 0; for (int i = st - 1; i; --i) { ans += a[i]; if (ans > 0) ++z[ans]; else ++f[-ans]; } ans = 0; int cnt = 1 + f[0]; ++f[0]; for (int i = st + 1; i <= n; ++i) { ans += a[i]; if (ans >= 0) cnt += f[ans]; else cnt += z[-ans]; } write(cnt), putchar(10); return 0; }