bzoj2653 主席树 【附赠数据生成器】

题目分析

要想要中位数尽可能大，就要让大的数尽量多，小的数尽量少。
而这个大小是相对于中位数而言的，所以我们可以用二分的方法先确定这个中位数x。然后，将所有大于等于x的数视作1,将所有小于x的数视作-1,这样，假若我们在满足条件的情况下可以找到一个区间和大于等于0的区间，说明可以增大x的值。否则，只能减小x的值。
那么，怎么确定区间呢？可以考虑对于原序列的每一个x都整一个1或-1的序列出来，然后搞一棵线段树，维护区间和，区间左端最大子区间和，区间右端最大子区间和三个值，这样每次查询[b+1,c-1]的区间和，[a,b]的区间右端最大子区间和和[c,d]的区间左端最大子区间和就可以检验了。
可是如果真的搞那么多线段树，空间受不了。这时候主席树出场了。

主席树是很多棵线段树构成的森林。

x的值从小到大排序后，每一棵线段树相对于上一棵线段树而言，都只有一条链上的值被从1改成了-1，所以主席树可以实现。

代码

注意数组从0开始编号

#include<bits/stdc++.h>
using namespace std;
#define zl l,r,s,mid,tr[x].ls
#define zr l,r,mid+1,t,tr[x].rs
const int N=20005;
int n,m,cur,ans,id[N],a[N],rt[N],q[5];
struct node{int ls,rs,sum,lx,rx;}tr[N*40];
void up(int x) {
	tr[x].sum=tr[tr[x].ls].sum+tr[tr[x].rs].sum;
	tr[x].lx=max(tr[tr[x].ls].lx,tr[tr[x].ls].sum+tr[tr[x].rs].lx);
	tr[x].rx=max(tr[tr[x].rs].rx,tr[tr[x].rs].sum+tr[tr[x].ls].rx);
}
int getsum(int l,int r,int s,int t,int x) {
	if(l<=s&&t<=r) return tr[x].sum;
	int mid=(s+t)>>1,re=0;
	if(l<=mid) re=getsum(zl);
	if(mid+1<=r) re+=getsum(zr);
	return re;
}
int getlx(int l,int r,int s,int t,int x) {
	if(l<=s&&t<=r) return tr[x].lx;
	int mid=(s+t)>>1;
	if(r<=mid) return getlx(zl);
	if(mid+1<=l) return getlx(zr);
	return max(getlx(zl),getsum(zl)+getlx(zr));
}
int getrx(int l,int r,int s,int t,int x) {
	if(l<=s&&t<=r) return tr[x].rx;
	int mid=(s+t)>>1;
	if(r<=mid) return getrx(zl);
	if(mid+1<=l) return getrx(zr);
	return max(getrx(zr),getsum(zr)+getrx(zl));
}
void build(int s,int t,int &x) {
	x=++cur,tr[x].sum=tr[x].lx=tr[x].rx=1;
	if(s==t) return;
	int mid=(s+t)>>1;
	build(s,mid,tr[x].ls),build(mid+1,t,tr[x].rs);
	up(x);
}
void ins(int s,int t,int &x,int y,int wz) {
	x=++cur,tr[x]=tr[y];
	if(s==t) {tr[x].sum=tr[x].lx=tr[x].rx=-1;return;}
	int mid=(s+t)>>1;
	if(wz<=mid) ins(s,mid,tr[x].ls,tr[y].ls,wz);
	else ins(mid+1,t,tr[x].rs,tr[y].rs,wz);
	up(x);
}
int check(int mid,int a,int b,int c,int d) {
	int re=0;
	if(b+1<=c-1) re+=getsum(b+1,c-1,0,n,rt[mid]);
	re+=getrx(a,b,0,n,rt[mid])+getlx(c,d,0,n,rt[mid]);
	return re>=0;
}
bool cmp(int x,int y) {return a[x]<a[y];}
int main()
{
	scanf("%d",&n);--n;
	for(int i=0;i<=n;++i) id[i]=i,scanf("%d",&a[i]);
	sort(id,id+1+n,cmp),build(0,n,rt[0]);
	for(int i=1;i<=n;++i) ins(0,n,rt[i],rt[i-1],id[i-1]);
	scanf("%d",&m);
	while(m--) {
		scanf("%d%d%d%d",&q[0],&q[1],&q[2],&q[3]);
		for(int i=0;i<=3;++i) q[i]=(q[i]+ans)%(n+1);
		sort(q,q+4);int l=0,r=n,mid;
		while(l<=r) {
			mid=(l+r)>>1;
			if(check(mid,q[0],q[1],q[2],q[3])) ans=mid,l=mid+1;
			else r=mid-1;
		}
		ans=a[id[ans]],printf("%d\n",ans);
	}
	return 0;
}

数据生成器

这东西应该没有什么用╮（╯＿╰）╭
本蒟蒻一开始把数组从1开始编号WA了一发，就搞了这东西出来对拍。

#include<bits/stdc++.h>
using namespace std;
int n,m;
int main()
{
    srand(time(0));
    n=rand()%10+1;printf("%d\n",n);
    for(int i=1;i<=n;++i) printf("%d ",rand()%10+1);
    puts("");
    m=rand()%10+1;printf("%d\n",m);
    for(int i=1;i<=m;++i)
    	printf("%d %d %d %d\n",rand()%10+1,rand()%10+1,rand()%10+1,rand()%10+1);
    return 0;
}