zoukankan      html  css  js  c++  java
  • BZOJ 1146 二分+链剖+线段树+treap

    思路:
    恶心的数据结构题……

    首先 我们 链剖 把树 变成序列 再 套一个 区间 第K大就好了……

    复杂度(n*log^4n)

    //By SiriusRen
    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    using namespace std;
    #define N 88888
    #define inf 100000000
    int n,q,first[N],next[N*2],v[N*2],t[N],tot,op,xx,yy;
    int fa[N],son[N],deep[N],top[N],siz[N],cnt,ch[N];
    int root[N*40],size;
    struct Treap{int ch[2],sz,cnt,v,rnd;}tr[N*40];
    void Upd(int k){tr[k].sz=tr[tr[k].ch[0]].sz+tr[tr[k].ch[1]].sz+tr[k].cnt;}
    void Rot(int &k,bool f){int t=tr[k].ch[f];tr[k].ch[f]=tr[t].ch[!f],tr[t].ch[!f]=k,Upd(k),Upd(t),k=t;}
    void Insert(int &k,int num){
        if(!k){k=++size;tr[k].sz=tr[k].cnt=1,tr[k].rnd=rand();tr[k].v=num;return;}
        tr[k].sz++;
        if(tr[k].v==num){tr[k].cnt++;return;}
        bool f=num>tr[k].v;
        Insert(tr[k].ch[f],num);
        if(tr[tr[k].ch[f]].rnd<tr[k].rnd)Rot(k,f);
    }
    void Del(int &k,int num){
        if(tr[k].v==num){
            if(tr[k].cnt>1)tr[k].cnt--,tr[k].sz--;
            else if(tr[k].ch[0]*tr[k].ch[1]==0)k=max(tr[k].ch[0],tr[k].ch[1]);
            else Rot(k,tr[tr[k].ch[0]].rnd>tr[tr[k].ch[1]].rnd),Del(k,num);
        }
        else tr[k].sz--,Del(tr[k].ch[num>tr[k].v],num);
    }
    int get_rank(int k,int num){
        if(!k)return 0;
        if(tr[k].v==num)return tr[tr[k].ch[1]].sz;
        else if(tr[k].v<num)return get_rank(tr[k].ch[1],num);
        else return get_rank(tr[k].ch[0],num)+tr[tr[k].ch[1]].sz+tr[k].cnt;
    }
    void insert(int l,int r,int pos,int num,int wei){
        Insert(root[pos],wei);
        if(l==r)return;
        int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
        if(mid<num)insert(mid+1,r,rson,num,wei);
        else insert(l,mid,lson,num,wei);
    }
    void change(int l,int r,int pos,int num,int wei){
        Del(root[pos],t[xx]),Insert(root[pos],wei);
        if(l==r)return;
        int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
        if(mid<num)change(mid+1,r,rson,num,wei);
        else change(l,mid,lson,num,wei);
    }
    int query(int l,int r,int pos,int L,int R,int num){
        if(l>=L&&r<=R)return get_rank(root[pos],num);
        int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
        if(mid<L)return query(mid+1,r,rson,L,R,num);
        else if(mid>=R)return query(l,mid,lson,L,R,num);
        else return query(l,mid,lson,L,R,num)+query(mid+1,r,rson,L,R,num);
    }
    void Add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
    void add(int x,int y){Add(x,y),Add(y,x);}
    void dfs(int x){
        siz[x]=1;
        for(int i=first[x];~i;i=next[i])
            if(v[i]!=fa[x]){
                fa[v[i]]=x,deep[v[i]]=deep[x]+1;
                dfs(v[i]),siz[x]+=siz[v[i]];
                if(siz[v[i]]>siz[son[x]])son[x]=v[i];
            }
    }
    void dfs2(int x,int tp){
        top[x]=tp,ch[x]=++cnt;
        insert(1,n,1,cnt,t[x]);
        if(son[x])dfs2(son[x],tp);
        for(int i=first[x];~i;i=next[i])
            if(v[i]!=fa[x]&&v[i]!=son[x])
                dfs2(v[i],v[i]);
    }
    int find(int x,int y,int num){
        int fx=top[x],fy=top[y],tmp=0;
        while(fx!=fy){
            if(deep[fx]<deep[fy])swap(fx,fy),swap(x,y);
            tmp+=query(1,n,1,ch[fx],ch[x],num);
            x=fa[fx],fx=top[x];
        }
        if(deep[x]>deep[y])swap(x,y);
        return tmp+query(1,n,1,ch[x],ch[y],num);
    }
    void b_srch(){
        int l=0,r=inf,ans;
        while(l<=r){
            int mid=(l+r)>>1;
            if(find(xx,yy,mid)>=op)l=mid+1;
            else ans=mid,r=mid-1;
        }
        if(!ans)puts("invalid request!");
        else printf("%d
    ",ans);
    }
    int main(){
        memset(first,-1,sizeof(first));
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++)scanf("%d",&t[i]);
        for(int i=1;i<n;i++)scanf("%d%d",&xx,&yy),add(xx,yy);
        dfs(1),dfs2(1,1);
        for(int i=1;i<=q;i++){
            scanf("%d%d%d",&op,&xx,&yy);
            if(op)b_srch();
            else change(1,n,1,ch[xx],yy),t[xx]=yy;
        }
    }

    这里写图片描述

  • 相关阅读:
    POJ--2112--Optimal Milking【Floyd+Dinic+二分答案】
    HTML5之WEB Storage
    java多线程具体总结
    CSU 1541 There is No Alternative (最小生成树+枚举)
    Blade
    数据模型封装演示样例
    Java中接口和抽象类的比較
    句子逆序
    王立平--split字符串切割
    UVA 12538 Version Controlled IDE 解题报告
  • 原文地址:https://www.cnblogs.com/SiriusRen/p/6532148.html
Copyright © 2011-2022 走看看