首页 > 代码库 > BZOJ1303 [CQOI2009]中位数图

BZOJ1303 [CQOI2009]中位数图

本文版权归ljh2000和博客园共有,欢迎转载,但须保留此声明,并给出原文链接,谢谢合作。

 

 

 

本文作者:ljh2000   

作者博客:http://www.cnblogs.com/ljh2000-jump/
转载请注明出处,侵权必究,保留最终解释权!

 

 

Description

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

Input

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

Output

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

Sample Input

7 4
5 7 2 4 3 1 6

Sample Output

4

HINT

第三个样例解释:{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000

 
 
正解:乱搞
解题报告:
  把原数组$change$一下,如果大于$b$则标为$1$,小于则标为$-1$,等于标为$0$。直接记录一下前缀和,如果和前面的某个前缀和相减是$0$则说明可行。注意序列必须跨越$b$,而且记录的前缀和在$b$之后就不应该插入了!
 
//It is made by ljh2000
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long LL;
const int MAXN = 100011;
int n,k,a[MAXN],tot,root;
int mp1[MAXN*2],mp2[MAXN*2];
LL ans;

inline int getint(){
    int w=0,q=0; char c=getchar(); while((c<‘0‘||c>‘9‘) && c!=‘-‘) c=getchar();
    if(c==‘-‘) q=1,c=getchar(); while (c>=‘0‘&&c<=‘9‘) w=w*10+c-‘0‘,c=getchar(); return q?-w:w;
}

inline void work(){
	n=getint(); k=getint(); for(int i=1;i<=n;i++) a[i]=getint();
	for(int i=1;i<=n;i++) if(a[i]>k) a[i]=1; else if(a[i]<k) a[i]=-1; else a[i]=0,root=i/*!!!*/;
	int ZERO=100000; mp2[ZERO]=1; tot=0; 
	for(int i=1;i<=n;i++) {
		tot+=a[i]; 
		if(i&1) {
			if(i>=root) ans+=mp2[ZERO+tot]/*!!!*/;
			else mp1[ZERO+tot]++;
		}
		else {
			if(i>=root) ans+=mp1[ZERO+tot]/*!!!*/;
			else mp2[ZERO+tot]++;
		}
	}
	printf("%lld",ans);
}

int main()
{
    work();
    return 0;
}

  

BZOJ1303 [CQOI2009]中位数图