bzoj 2653: middle 可持久化线段树+二分答案

题意

一个长度为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;
}