zoukankan      html  css  js  c++  java
  • BZOJ 3123 主席树 启发式合并

    思路:
    主席树 搞树上的k大 x+y-lca(x,y)-fa(lca(x,y))
    按照size小树往大树上插 启发式合并
    n*log^2n的 搞定~

    //By SiriusRen
    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    using namespace std;
    #define N 666666
    const int inf=1000000000;char op[2];
    int cases,n,m,t,a[N],first[N],next[N],v[N],w[N],tot,ans;
    int xx,yy,zz,jy,f[N],wei[N],size[N],fa[N][22],deep[N],root[N],cnt;
    int tree[N*50],lson[N*50],rson[N*50];
    int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
    void merge(int x,int y){int fx=find(x),fy=find(y);f[fx]=fy,size[fy]+=size[fx];}
    void add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
    void insert(int x,int &y,int l,int r,int val){
        if(!y)y=++cnt;
        tree[y]=tree[x]+1;
        if(l==r)return;
        int mid=(l+r)>>1;
        if(val<=mid)rson[y]=rson[x],insert(lson[x],lson[y],l,mid,val);
        else lson[y]=lson[x],insert(rson[x],rson[y],mid+1,r,val);
    }
    void dfs(int x){
        for(int i=1;i<=19;i++)fa[x][i]=fa[fa[x][i-1]][i-1];
        deep[x]=deep[fa[x][0]]+1,root[x]=0;
        insert(root[fa[x][0]],root[x],0,inf,wei[x]);
        for(int i=first[x];~i;i=next[i])if(v[i]!=fa[x][0])
            fa[v[i]][0]=x,dfs(v[i]);
    }
    int lca(int x,int y){
        if(deep[x]<deep[y])swap(x,y);
        for(int i=19;~i;i--)if(deep[fa[x][i]]>=deep[y])x=fa[x][i];
        if(x==y)return x;
        for(int i=19;~i;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
        return fa[x][0];
    }
    int kth(int x,int y,int lca,int falca,int l,int r,int k){
        if(l==r)return l;
        int mid=(l+r)>>1;
        int temp=tree[lson[x]]+tree[lson[y]]-tree[lson[lca]]-tree[lson[falca]];
        if(k<=temp)return kth(lson[x],lson[y],lson[lca],lson[falca],l,mid,k);
        else return kth(rson[x],rson[y],rson[lca],rson[falca],mid+1,r,k-temp);
    }
    int main(){
        scanf("%d",&cases);
        memset(first,-1,sizeof(first)),tot=ans=cnt=0;
        scanf("%d%d%d",&n,&m,&t);
        for(int i=1;i<=n;i++)scanf("%d",&wei[i]),f[i]=i,size[i]=1;
        for(int i=1;i<=m;i++)
            scanf("%d%d",&xx,&yy),
            add(xx,yy),add(yy,xx),merge(xx,yy);
        for(int i=1;i<=n;i++)if(!fa[i][0])dfs(i);
        for(int i=1;i<=t;i++){
            scanf("%s%d%d",op,&xx,&yy),xx^=ans,yy^=ans;
            if(op[0]=='Q')
                scanf("%d",&zz),zz^=ans,jy=lca(xx,yy),
                printf("%d
    ",ans=kth(root[xx],root[yy],root[jy],root[fa[jy][0]],0,inf,zz));
            else{
                int fx=find(xx),fy=find(yy);
                if(size[fx]>size[fy])swap(xx,yy);
                add(xx,yy),add(yy,xx),merge(xx,yy),fa[xx][0]=yy,dfs(xx);
            }
        }
    }

    这里写图片描述

  • 相关阅读:
    HDU 3579 Hello Kiki 中国剩余定理
    DHU 1788 Chinese remainder theorem again 中国剩余定理
    初学--求解模线性方程组(中国余数定理)。
    山东第四届省赛: Boring Counting 线段树
    山东第四届省赛C题: A^X mod P
    福州大学oj 1752 A^B mod C ===>数论的基本功。位运用。五星*****
    HDU 1576 A/B 暴力也能过。扩展欧几里得
    POJ 1061 青蛙的约会 扩展欧几里德--解不定方程
    HDU 2669 Romantic 扩展欧几里德---->解不定方程
    南阳nyoj 56 阶乘因式分解(一)
  • 原文地址:https://www.cnblogs.com/SiriusRen/p/6532099.html
Copyright © 2011-2022 走看看