bzoj 3196 Tyvj 1730 二逼平衡树【线段树 套 splay】

四舍五入就是个暴力。
对于线段树的每个区间都开一棵按权值排序的splay
对于第二个操作，二分一下，每次查询mid的排名，复杂度 $ O(nlog(n)^{3}) $
其余的操作都是$ O(nlog(n)^{2}) $

#include<iostream>
#include<cstdio>
using namespace std;
const int N=50005,T=2500250,inf=1e9;
int n,m,a[N],rt[T],tot,mn=inf,mx=-inf;
struct pinghengshu
{
    int c[2],v,s,f;
}t[T];
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p=='-')
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void pu(int x)
{
    t[x].s=t[t[x].c[0]].s+t[t[x].c[1]].s+1;
}
void zhuan(int p)
{
    int q=t[p].f,y=t[q].f,x=(t[q].c[1]==p);
    t[q].c[x]=t[p].c[x^1];t[t[q].c[x]].f=q;
    t[p].c[x^1]=q;t[q].f=p;t[p].f=y;
    if(y)
    {
        if(t[y].c[0]==q)
            t[y].c[0]=p;
        else 
            t[y].c[1]=p;
    }
    pu(q);
    pu(p);
}
void splay(int ro,int x)
{
    for(int y;y=t[x].f;zhuan(x))
        if(t[y].f)
        {
            if((t[y].c[0]==x)^(t[t[y].f].c[0]==y))
                zhuan(y);
            else
                zhuan(x);
        }
    rt[ro]=x;
}
void charu(int ro,int x,int z)
{
    t[x].s++;
    while(t[x].c[t[x].v<z])
        x=t[x].c[t[x].v<z],t[x].s++;
    t[x].c[t[x].v<z]=++tot;
    t[tot].f=x;
    t[tot].s=1;
    t[tot].v=z;
    splay(ro,tot);
}
void build(int ro,int l,int r)
{
    rt[ro]=++tot;
    t[tot].s=1;
    t[tot].v=a[l];
    if(l==r)
        return;
    for(int i=l+1;i<=r;i++)
        charu(ro,rt[ro],a[i]);
    int mid=(l+r)>>1;
    build(ro<<1,l,mid);
    build(ro<<1|1,mid+1,r);
}
int zhao(int x,int z)
{
    if(t[x].v==z)
        return x;
    while(t[x].c[t[x].v<z])
    {
        x=t[x].c[t[x].v<z];
        if(t[x].v==z)
            return x;
    }
    return 0;
}
void shanchu(int ro,int x)
{
    splay(ro,x);
    if(!t[x].c[0])
        rt[ro]=t[x].c[1],t[t[x].c[1]].f=0;
    else if(!t[x].c[1])
        rt[ro]=t[x].c[0],t[t[x].c[0]].f=0;
    else
    {
        t[t[x].c[0]].f=0;
        int tmp=t[x].c[0];
        while(t[tmp].c[1])
            tmp=t[tmp].c[1];
        splay(ro,tmp);
        t[tmp].c[1]=t[x].c[1];
        t[t[x].c[1]].f=tmp;
        pu(tmp);
    }
}
int fd(int x,int z)
{
    int ans=0;
    while(x)    
    {
        if(t[x].v>z)
            x=t[x].c[0];
        else 
            ans+=t[t[x].c[0]].s+1,x=t[x].c[1];
   }
   return ans;
}
int paiming(int x,int l,int r,int ll,int rr,int kk)
{
    if(ll<=l&&rr>=r)
        return fd(rt[x],kk);
    int mid=(l+r)>>1,ans=0;
    if(ll<=mid)
        ans+=paiming(x<<1,l,mid,ll,rr,kk);
    if(rr>mid)
        ans+=paiming(x<<1|1,mid+1,r,ll,rr,kk);
    return ans;
}
void gai(int x,int l,int r,int p,int z)
{
    charu(x,rt[x],z);
    int x1=zhao(rt[x],a[p]);
    shanchu(x,x1);
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    if(p<=mid)
        gai(x<<1,l,mid,p,z);
    else 
        gai(x<<1|1,mid+1,r,p,z);
}
int pre(int x,int z)
{
    int ans=-inf;
    while(t[x].c[t[x].v<=z])
    {
        if(t[x].v<=z)
            ans=t[x].v;
        x=t[x].c[t[x].v<=z];
    }
    if(t[x].v<=z)
        ans=max(ans,t[x].v);
    return ans;
}
int suc(int x,int z)
{
    int ans=inf;
    while(t[x].c[t[x].v<z])
    {
        if(t[x].v>=z)
            ans=t[x].v;
        x=t[x].c[t[x].v<z];
    }
    if(t[x].v>=z)
        ans=min(ans,t[x].v);
    return ans;
}
int qianqu(int x,int l,int r,int ll,int rr,int kk)
{
    if(l>=ll&&r<=rr)
        return pre(rt[x],kk);
    int ans=-inf;
    int mid=(l+r)>>1;
    if(ll<=mid)
        ans=max(ans,qianqu(x<<1,l,mid,ll,rr,kk));
    if(rr>mid)
        ans=max(ans,qianqu(x<<1|1,mid+1,r,ll,rr,kk));
    return ans;
}
int houji(int x,int l,int r,int ll,int rr,int kk)
{
    if(l>=ll&&r<=rr)
        return suc(rt[x],kk);
    int ans=inf;
    int mid=(l+r)>>1;
    if(ll<=mid)
        ans=min(ans,houji(x<<1,l,mid,ll,rr,kk));
    if(rr>mid)
        ans=min(ans,houji(x<<1|1,mid+1,r,ll,rr,kk));
    return ans;
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++)
        a[i]=read(),mx=max(a[i],mx),mn=min(mn,a[i]);
    build(1,1,n);
    while(m--)
    {
        int o=read();
        if(o==1)
        {//cout<<"paiming"<<endl;
            int l=read(),r=read(),k=read();
            printf("%d\n",paiming(1,1,n,l,r,k-1)+1);
        }
        else if(o==2)
        {//cout<<"shu"<<endl;
            int ll=read(),rr=read(),k=read(),l=mn,r=mx;
            while(l<=r)
            {
                int mid=(l+r)>>1;//cout<<mid<<endl;
                if(paiming(1,1,n,ll,rr,mid)<k)
                    l=mid+1;
                else
                    r=mid-1;
            }
            printf("%d\n",l);
        }
        else if(o==3)
        {//cout<<"gai"<<endl;
            int p=read(),k=read();
            gai(1,1,n,p,k);
            a[p]=k;
        }
        else if(o==4)
        {//cout<<"qianqu"<<endl;
            int l=read(),r=read(),k=read();
            printf("%d\n",qianqu(1,1,n,l,r,k-1));
        }
        else
        {//cout<<"houzhui"<<endl;
            int l=read(),r=read(),k=read();
            printf("%d\n",houji(1,1,n,l,r,k+1));
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/lokiii/p/8142677.html