【NOIP 模拟题】中位数（规律+递推）

中位数（median.cpp）

【题目描述】
    有一个长度为N的数列{A1,A2,...An}，这N和数字恰好是1..N的一个排列。你需要统计有多少个子序列{Ai,Ai+1,...Aj}满足：i<=j且j-i+1为奇数，序列的中位数为B。例如{5,1,3}的中位数为3。
【输入文件】
    第一行包含两个正整数N和B；第二行包含N个整数，第i个整数为Ai。
【输出文件】
    仅包含一个整数，为满足条件的子序列的个数。
【样例输入】
    7 4
    5 7 2 4 3 1 6
【样例输出】
    4
【数据规模】
    对于30%的数据中，满足N<=100；
    对于60%的数据中，满足N<=1000；

    对于100%的数据中，满足N<=100000,1<=B<=N。

——————————————————————————————————

【题解】【规律+递推】

【因为B是序列的中位数，所以B一定存在于序列中，所以，记录B在序列中的位置num，并把相对应的数置为0；把小于B的数置为-1；大于B的数置为1。】

【分别从num向左向右扫描，并分别设一个累加器，将枚举时遇到的0、1、-1都加入累加器里，并分别用L、R数组记录累加器中每个数出现的次数】

【最后求：∑L[i]*R[-i]（但因为C++没有负数下标，所以要整体向右移n个下标）】

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a[100010],L[200010],R[200010],n,b,num,ans;
int main()
{
	freopen("median.in","r",stdin);
	freopen("median.out","w",stdout);
	int i,j;
	scanf("%d%d",&n,&b);
	for(i=1;i<=n;++i)
	 {
	 	scanf("%d",&a[i]);
	 	if(a[i]==b) {num=i; a[i]=0; continue;}
	 	if(a[i]>b) a[i]=1;
	 	 else a[i]=-1;
	 }
	int l=n,r=n;
	for(i=num;i>0;--i) l+=a[i],L[l]++;
	for(i=num;i<=n;++i) r+=a[i],R[r]++;
	int len=2*n;
	for(i=0;i<=len;++i)
	  ans+=(L[i]*R[len-i]);
	printf("%d\n",ans);
	return 0;
}