[CQOI2009]中位数图

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

复制7 4 5 7 2 4 3 1 6

7 4
5 7 2 4 3 1 6

输出

复制4

4

备注:

对于30%的数据中，满足 n≤100n \le 100n≤100；

对于60%的数据中，满足 n≤1000n \le 1000n≤1000；

对于100%的数据中，满足 n≤100000,1≤b≤nn \le 100000,1 \le b \le nn≤100000,1≤b≤n。

13:42

前置知识：前缀和

题解：首先简化转化问题，对于中位数，其他数是什么怎么排序不重要，重要的是中位数其实就是大于他的数和小于他的数相同，所以计算包含b的所有连续区域中大于b和小于b的数目相同的区域个数。但计算枚举所有包含b的连续子区域数据规模很大大概是o（n^2）任然可能超时，所以继续优化枚举连续区域的时间，可以发现包含b的连续区域左区域大于b的个数等于右区域小于b的个数或者左区域小于b的个数等于右区域大于b的个数则b就是中位数，按照这个特性去优化。

 

#include <iostream>
#include <unordered_map>
using namespace std;
int n, a[100000], d[100000] = { 0 }, map[300000] = { 0 };
int main() {
    int b;
    cin >> n >> b;
    int index = 0;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
      ""把大于b的数字记为1，小于b的数字记为-1，等于b的数的记为0""
        if (a[i] > b)
        {
            a[i] = 1;
        }
        else if (a[i] < b)
        {
            a[i] = -1;
        }
        else
        {
            index = i;//记录b的位置
            a[i] = 0;
        }
    }
    int ans = 1;
   ""记录左区域大于b或小于b不同个数的个数""
    for (int i = index - 1; i >= 1; i--)
    {
        d[i] = d[i + 1] + a[i];//左边前缀和
        map[n + d[i]]++;//防止越界
        if (d[i] == 0)ans++;//分类讨论0特殊值的处理
    }
    for (int i = index + 1; i <= n; i++)
    {
        d[i] = d[i - 1] + a[i];//右边前缀和
        ans += map[n - d[i]];//取d[i]的相反数
        if (d[i] == 0)ans++;//分类讨论0特殊值的处理
    }
    cout << ans << endl;
}
// 64 位输出请用 printf("%lld")