码迷,mamicode.com
首页 > 其他好文 > 详细

主席树+启发式合并(LT) BZOJ3123

时间:2015-05-15 17:23:20      阅读:549      评论:0      收藏:0      [点我收藏+]

标签:

好久没做题了,写道SBT又RE又T

查询:主席树裸题。

修改:对于两个树合并重建小的树。

注意fa[x][i]重新计算时要清空

技术分享
#include<cstdio> 
#include<cctype> 
#include<cstring> 
#include<algorithm> 
using namespace std; 
inline int read() 
{ 
    char c=getchar();int x=0,sig=1; 
    for(;!isdigit(c);c=getchar()) if(c==-) sig=-1; 
    for(;isdigit(c);c=getchar()) x=x*10+c-0; 
    return x*sig; 
} 
inline void print(int x) 
{ 
    int buf[10],l=0; 
    if(!x) l++; 
    while(x) buf[++l]=x%10,x/=10; 
    for(;l;l--) putchar(buf[l]+0); 
    putchar(\n); 
} 
const int maxn=80010; 
const int maxnode=10000010; 
int ls[maxnode],rs[maxnode],s[maxnode],ToT; 
void update(int& y,int x,int l,int r,int pos) 
{ 
     s[y=++ToT]=s[x]+1;if(l==r) return; 
     int mid=l+r>>1;ls[y]=ls[x];rs[y]=rs[x]; 
     if(pos<=mid) update(ls[y],ls[x],l,mid,pos); 
     else update(rs[y],rs[x],mid+1,r,pos); 
} 
int query(int x1,int x2,int y1,int y2,int l,int r,int k) 
{ 
    if(l==r) return l; 
    int mid=l+r>>1,k2=s[ls[y1]]+s[ls[y2]]-s[ls[x1]]-s[ls[x2]]; 
    if(k<=k2) return query(ls[x1],ls[x2],ls[y1],ls[y2],l,mid,k); 
    return query(rs[x1],rs[x2],rs[y1],rs[y2],mid+1,r,k-k2); 
} 
int n,m,q,val[maxn],tmp[maxn],root[maxn],lastans; 
int first[maxn],next[maxn*2],to[maxn*2],e,fa[maxn][20],dep[maxn]; 
int pa[maxn],sz[maxn];
void AddEdge(int a,int b) 
{ 
    to[++e]=b;next[e]=first[a];first[a]=e; 
    to[++e]=a;next[e]=first[b];first[b]=e; 
} 
void dfs(int x,int f) 
{ 
    dep[x]=dep[fa[x][0]=f]+1;update(root[x],root[f],1,n,val[x]);
    sz[pa[x]]++;
    for(int i=1;i<17;i++) fa[x][i]=fa[fa[x][i-1]][i-1]; 
    for(int i=first[x];i;i=next[i]) if(to[i]!=f) pa[to[i]]=pa[x],dfs(to[i],x); 
} 
int lca(int p,int q) 
{ 
    if(dep[p]<dep[q]) swap(p,q);
    for(int i=19;i>=0;i--) if((1<<i)<=dep[p]-dep[q]) p=fa[p][i];
    for(int i=19;i>=0;i--) if(fa[p][i]!=fa[q][i]) p=fa[p][i],q=fa[q][i];
    return p==q?p:fa[p][0];
}
int main() 
{ 
    read();n=read();m=read();q=read(); 
    for(int i=1;i<=n;i++) tmp[i]=val[i]=read();
    sort(tmp+1,tmp+n+1); 
    for(int i=1;i<=n;i++) val[i]=lower_bound(tmp+1,tmp+n+1,val[i])-tmp; 
    for(int i=0;i<m;i++) AddEdge(read(),read());
    for(int i=1;i<=n;i++) if(!fa[i][0]) pa[i]=i,dfs(i,0); 
    for(int i=0;i<q;i++) 
    { 
        char tp=getchar(); 
        while(!isalpha(tp)) tp=getchar(); 
        if(tp==Q) 
        { 
            int x=read()^lastans,y=read()^lastans,k=read()^lastans; 
            int c=lca(x,y); 
            print(lastans=tmp[query(root[c],root[fa[c][0]],root[x],root[y],1,n,k)]); 
        }
        else
        { 
            int u=read()^lastans,v=read()^lastans; 
            int x=pa[u],y=pa[v]; 
            if(sz[x]>sz[y]) swap(x,y),swap(u,v); 
            pa[u]=y;AddEdge(u,v);dfs(u,v); 
        } 
    } 
    return 0; 
}
View Code

 

主席树+启发式合并(LT) BZOJ3123

标签:

原文地址:http://www.cnblogs.com/wzj-is-a-juruo/p/4506292.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!