题意
一个长度为n的序列a,设其排过序之后为b,其中位数定义为b[n/2],其中a,b从0开始标号,除法取下整。给你一个长度为n的序列s。回答Q个这样的询问:s的左端点在[a,b]之间,右端点在[c,d]之间的子序列中,最大的中位数。其中a
分析
据说这是clj的题,果然牛逼。。。一开始想到了一个主席树+分块的做法,推了一下发现时间复杂度是错的。。。
首先我们可以二分答案,设为x,然后把不小于x的数设为1,小于x的数设为-1,如果最大的序列和不小于0,则x是合法的,反之则不合法。
问题在于如何快速求出最大的序列和。
我们可以以元素为下标建主席树,也就是从小到大加入元素。最小元素的线段树中每个位置都为1。每到一个元素,就把上一个元素的位置在这个元素的主席树中的位置标程-1,同时维护一下区间和,左边最大值和右边最大值即可。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=20005;
int n,sz,a[N],w[N],root[N];
vector<int> vec[N];
struct tree{int l,r,ls,rs,s;}t[N*100];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void build(int &d,int l,int r)
{
d=++sz;t[d].ls=t[d].rs=t[d].s=r-l+1;
if (l==r) return;
int mid=(l+r)/2;
build(t[d].l,l,mid);build(t[d].r,mid+1,r);
}
void updata(int d)
{
t[d].s=t[t[d].l].s+t[t[d].r].s;
t[d].ls=max(t[t[d].l].ls,t[t[d].l].s+t[t[d].r].ls);
t[d].rs=max(t[t[d].r].rs,t[t[d].r].s+t[t[d].l].rs);
}
void ins(int &d,int l,int r,int x)
{
int p=d;d=++sz;t[d]=t[p];
if (l==r) {t[d].ls=t[d].rs=t[d].s=-1;return;}
int mid=(l+r)/2;
if (x<=mid) ins(t[d].l,l,mid,x);
else ins(t[d].r,mid+1,r,x);
updata(d);
}
int get_s(int d,int l,int r,int x,int y)
{
if (x>y) return 0;
if (l==x&&r==y) return t[d].s;
int mid=(l+r)/2;
return get_s(t[d].l,l,mid,x,min(y,mid))+get_s(t[d].r,mid+1,r,max(x,mid+1),y);
}
pair<int,int> get_ls(int d,int l,int r,int x,int y)
{
if (l==x&&r==y) return make_pair(t[d].s,t[d].ls);
int mid=(l+r)/2;
if (y<=mid) return get_ls(t[d].l,l,mid,x,y);
else if (x>mid) return get_ls(t[d].r,mid+1,r,x,y);
else
{
pair<int,int> u=get_ls(t[d].l,l,mid,x,mid),v=get_ls(t[d].r,mid+1,r,mid+1,y);
return make_pair(u.first+v.first,max(u.second,u.first+v.second));
}
}
pair<int,int> get_rs(int d,int l,int r,int x,int y)
{
if (l==x&&r==y) return make_pair(t[d].s,t[d].rs);
int mid=(l+r)/2;
if (y<=mid) return get_rs(t[d].l,l,mid,x,y);
else if (x>mid) return get_rs(t[d].r,mid+1,r,x,y);
else
{
pair<int,int> u=get_rs(t[d].l,l,mid,x,mid),v=get_rs(t[d].r,mid+1,r,mid+1,y);
return make_pair(u.first+v.first,max(v.second,v.first+u.second));
}
}
int main()
{
n=read();
for (int i=1;i<=n;i++) w[i]=a[i]=read();
sort(w+1,w+n+1);int w1=unique(w+1,w+n+1)-w-1;
for (int i=1;i<=n;i++) a[i]=lower_bound(w+1,w+w1+1,a[i])-w,vec[a[i]].push_back(i);
build(root[1],1,n);
for (int i=2;i<=w1;i++)
{
root[i]=root[i-1];
for (int j=0;j<vec[i-1].size();j++) ins(root[i],1,n,vec[i-1][j]);
}
int ans=0,q=read();
while (q--)
{
int h[4];
for (int i=0;i<4;i++) h[i]=(read()+ans)%n+1;
sort(h,h+4);
int a=h[0],b=h[1],c=h[2],d=h[3];
int l=1,r=w1;
while (l<=r)
{
int mid=(l+r)/2;
if (get_s(root[mid],1,n,b+1,c-1)+get_rs(root[mid],1,n,a,b).second+get_ls(root[mid],1,n,c,d).second>=0) l=mid+1;
else r=mid-1;
}
printf("%d\n",ans=w[l-1]);
}
return 0;
}