牛客OJ NC19913中位数图 解题报告

原题链接：

登录—专业IT笔试面试备考平台_牛客网

题目描述：

给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后，位于中间的数。

输入描述:

第一行为两个正整数n和b ，第二行为1~n 的排列。

输出描述:

输出一个整数，即中位数为b的连续子序列个数。

输入

7 4
5 7 2 4 3 1 6

输出

4

解题思路以及代码（CPP）：

#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1e5 + 10;
const int INF = 0x3fffffff;
const int mod = 1000000007;
int n, b;
int a[maxn];  // ai大于b则a[i]=1，ai小于b则a[i]=-1，ai等于b则为0
int s[maxn];  // 维护以ai==b为中心向两边扩散的前缀后缀和
map<int, int> cnt;

/*
    b如果要是中位数的话，大于他的数和小于他的数各输就需要一样，至于序列每个数字具体是多少，
    我们并不需要关心，我们可以把大于b的数记为1，小于b的数记为-1，等于他的数记为0，那么我们
    就将问题转化为区间和为0且里边有0的长度为奇数的区间个数（因为1和-1个数要一样，并且还有一个0，
    所以长度为奇数这个条件在前两个条件里就已经满足了）。
    我们可以以0作为起点，分别看他的左边和右边，计算从他开始的左边的后缀和，以及从他开始的右边的前缀和，
    显然当左边后缀和为x的时候，右边的前缀和要为-x才能保证区间和为0，所以在计算左边后缀和的时候，同时
    记录每个后缀和i的个数，在计算右边前缀和的时候进行统计。
*/

void solve() {
    cin >> n >> b;
    int idx = 0;
    // 预处理a
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        if (a[i] > b) {
            a[i] = 1;
        } else if (a[i] < b) {
            a[i] = -1;
        } else {
            a[i] = 0;
            idx = i;
        }
    }
    // 计算前后缀、统计
    int ans = 0;
    cnt[0]++;
    for (int i = idx - 1; i >= 1; i--) {
        s[i] = s[i + 1] + a[i];
        cnt[s[i]]++;
    }
    ans += cnt[0];
    for (int i = idx + 1; i <= n; i++) {
        s[i] = s[i - 1] + a[i];
        ans += cnt[-s[i]];
    }
    cout << ans << endl;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cout << fixed;
    cout.precision(18);

    solve();
    return 0;
}