题目大意:给出一个1~n的排列,问这其中的中位数为k的子序列的数量是多少。
思路:要想让k为中位数,一定要让它出现在这个序列中所以就以这个数为中心向两边拓展。想让一个数成为中位数,那么这个序列中出现的小于它的数一定要等于大于它的数。考虑到这个数左右都有可能有数字,那么就可以表示成:前多+后多=前少+后少,也就是前多-前少=后少-后多。之后就用个数组随意统计一下就行了。
CODE:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAX 100010
#define BASE 100001
using namespace std;
int points,pos,p;
int src[MAX];
int _less[MAX],more[MAX];
int cnt[MAX << 1];
int main()
{
cin >> points >> p;
for(int i = 1; i <= points; ++i) {
scanf("%d",&src[i]);
if(src[i] == p) pos = i;
}
++cnt[BASE];
for(int i = pos - 1; i; --i) {
_less[i] = _less[i + 1],more[i] = more[i + 1];
src[i] < p ? ++_less[i]:++more[i];
}
for(int i = pos + 1; i <= points; ++i) {
_less[i] = _less[i - 1],more[i] = more[i - 1];
src[i] < p? ++_less[i]:++more[i];
++cnt[BASE + _less[i] - more[i]];
}
int ans = 0;
for(int i = pos; i; --i)
ans += cnt[BASE + more[i] - _less[i]];
cout << ans << endl;
return 0;
}