题目描述
给出1~n的一个排列,统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后,位于中间的数。
输入描述:
第一行为两个正整数n和b ,第二行为1~n 的排列。
输出描述:
输出一个整数,即中位数为b的连续子序列个数。
输入
7 4
5 7 2 4 3 1 6
输出
4
链接:https://ac.nowcoder.com/acm/problem/19913
solution
b是中位数的奇数长度的连续子序列,大于b和小于b的个数是一样的,把大于b的设为1,小于b的设为-1,等于b设为0,这样就变成了找区间和为0,b左边求后缀和,b右边求前缀和,0和左边或右边的0都符合,左边的0和右边的0也可以,左边的和右边的相反数也可以。
对样例来说:
5 7 2 4 3 1 6(原序列)
1 1 -1 0 -1 -1 1(大的变1,小的变-1,等于的变0)
1 0 -1 0 -1 -2 -1(左边后缀和,右边前缀和)
code
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 +