【BZOJ2653】【二分法】【主席树】middle

题目大意：给出长度为n的一个序列s，回答q个询问：s的左端点在[a,b]，右端点在[c,d]的子序列中，最大的中位数，强制在线。

将问题转化为判定性问题，先二分出一个数x，然后将区间内大于等于x的数标记为1，小于x的数标记为-1

求区间和sum，如果sum >= 0说明中位数可以比当前数大，这样可以通过迭代求出最大的中位数。

但是题目中限定了左端点和右端点，不难想到可以通过求最大子序列的方法求出标记和。

但是如果我们暴力维护标记和最大子序列肯定会TLE，这时候就可以使用主席树来维护标记和最大子序列的和。

维护3个标记，分别表示区间和，最大左连续和，最大右连续和

一开始所有的标记都为1，然后将序列从小到大地插入主席树中，当插入第pos位时，第pos棵线段树的pos - 1位的标记就赋值为-1。

在询问x的时候就可以直接查询第root[x]棵线段树的值了。

至于用线段树维护最大子序列的和已经做过很多了，不再赘述。

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#define lson l,m,ls[rt]
#define rson m + 1,r,rs[rt]
using namespace std;
const int maxn = 200000 + 10;
const int maxm = 8005000;
struct Number
{
	int pos,num;
}p[maxn];
int root[maxn];
int ls[maxm],rs[maxm],lm[maxm],rm[maxm],sum[maxm];
int tot = 0;
int n,m;
int cmp(const Number &a,const Number &b)
{
	return a.num < b.num;
}

void init()
{
	freopen("bzoj2653.in","r",stdin);
	freopen("bzoj2653.out","w",stdout);
}

void push_up(int rt)
{
	sum[rt] = sum[ls[rt]] + sum[rs[rt]];
	lm[rt] = max(lm[ls[rt]],sum[ls[rt]] + lm[rs[rt]]);
	rm[rt] = max(rm[rs[rt]],sum[rs[rt]] + rm[ls[rt]]);
}

void build(int l,int r,int &rt)
{
	rt = ++tot;
	if(l == r)
	{
		lm[rt] = rm[rt] = sum[rt] = 1;
		return;
	}
	int m = (l + r) >> 1;
	build(lson);
	build(rson);
	push_up(rt);
}

int query_all(int L,int R,int l,int r,int rt)
{
	if(L == l && r == R)return sum[rt];
	int m = (l + r) >> 1;
	if(R <= m)return query_all(L,R,lson);
	else if(L > m)return query_all(L,R,rson);
	else return query_all(L,m,lson) + query_all(m + 1,R,rson);
}

int query_left(int L,int R,int l,int r,int rt)
{
	if(L == l && r == R)return lm[rt];
	int m = (l + r) >> 1;
	if(R <= m)return query_left(L,R,lson);
	else if(L > m)return query_left(L,R,rson);
	else return max(query_left(L,m,lson),query_all(L,m,lson) + query_left(m + 1,R,rson));
}

int query_right(int L,int R,int l,int r,int rt)
{
	if(L == l && r == R)return rm[rt];
	int m = (l + r) >> 1;
	if(R <= m)return query_right(L,R,lson);
	else if(L > m)return query_right(L,R,rson);
	else return max(query_right(m + 1,R,rson),query_all(m + 1,R,rson) + query_right(L,m,lson));
}

void update(int last,int l,int r,int pos,int val,int &rt)
{
	rt = ++tot;
	ls[rt] = ls[last],rs[rt] = rs[last];
	if(l == r)
	{
		sum[rt] = lm[rt] = rm[rt] = val;
		return;
	}
	int m = (l + r) >> 1;
	if(pos <= m)update(ls[last],l,m,pos,val,ls[rt]);
	else update(rs[last],m + 1,r,pos,val,rs[rt]);
	push_up(rt);
}

bool check(int k,int a,int b,int c,int d)
{
	int ret = 0;
	if(b + 1 < c)ret = query_all(b + 1,c - 1,0,n - 1,root[k]);
	ret += query_left(c,d,0,n - 1,root[k]);
	ret += query_right(a,b,0,n - 1,root[k]);
	return ret >= 0;
}

void readdata()
{
	scanf("%d",&n);
	for(int i = 0;i < n;i++)
	{
		p[i].pos = i;
		scanf("%d",&p[i].num);
	}
	stable_sort(p,p + n,cmp);
	build(0,n - 1,root[0]);
	for(int i = 1;i < n;i++)update(root[i-1],0,n - 1,p[i - 1].pos,-1,root[i]);
	scanf("%d",&m);
	int ans = 0,q[4];
	for(int i = 1;i <= m;i++)
	{
		for(int j = 0;j < 4;j++)scanf("%d",&q[j]);
		for(int j = 0;j < 4;j++)q[j] = (ans + q[j]) % n;
		stable_sort(q,q + 4);
		int l = 0,r = n - 1;
		while(l < r)
		{
			int m = (l + r + 1) >> 1;
			if(check(m,q[0],q[1],q[2],q[3]))l = m;
			else r = m - 1;
		}
		ans = p[l].num;
		printf("%d\n",ans);
	}
}

int main()
{
	init();
	readdata();
	return 0;
}