最近一直在被lrj的紫书的最后一章虐,前几天写LCT被卡常数卡成狗,现在来玩玩LCA。
以前用的都是Tarjan求LCA,只能处理离线问题(倍增老写错所以就懒得用了),今天学了下欧拉序列,随手下了个LCA,然后又写挂了。。。。。。写错了一个变量名这事不能怪我TAT。
其实利用欧拉序列求LCA的思想还是很简单的。
我们令pos(k)为节点k在欧拉序列中第一次出现的序号,再得到一个深度序列B,B[i]即为欧拉序列中序号为i的位置的节点的深度(定语好长),于是有LCA(u,v)=RMQ(B,pos(u),pos(v)),这里RMQ返回下标,然后就好说了。
不过由于我太沙茶,看不懂±1RMQ的O(n)构造方法(懒得看),于是就写了个O(nlgn)-O(1)版的。
例题自然是POJ1330(这么水的题完全不用LCA啊) 和 POJ1470(还算可以的标准LCA问题)
代码是POJ1470的。
#include<iostream> #include<cstdio> #include<cstring> using namespace std; const int N=900+10; struct Edge{int to,next;}e[N]; int head[N],cnt,n; int F[N*2],dep[N*2],sz,st[N*2][20],pos[N],ans[N]; void ins(int u,int v){ cnt++;e[cnt].to=v;e[cnt].next=head[u];head[u]=cnt; } void dfs(int u,int depth){ F[++sz]=u;pos[u]=sz;dep[sz]=depth; for(int i=head[u];i;i=e[i].next){ int v=e[i].to; dfs(v,depth+1); F[++sz]=u;dep[sz]=depth; } } void rmq_init(){ for(int i=1;i<n*2;i++)st[i][0]=i; for(int j=1;(1<<j)<n*2;j++) for(int i=1;i+(1<<j)-1<n*2;i++) if(dep[st[i][j-1]]<dep[st[i+(1<<(j-1))][j-1]])st[i][j]=st[i][j-1]; else st[i][j]=st[i+(1<<(j-1))][j-1]; } int rmq(int l,int r){ int k=0; while(1<<(k+1)<=r-l+1)k++; if(dep[st[l][k]]<dep[st[r-(1<<k)+1][k]])return st[l][k]; return st[r-(1<<k)+1][k]; } int LCA(int u,int v){ if(pos[u]>pos[v])swap(u,v); return F[rmq(pos[u],pos[v])]; } bool isroot[N]; int main(){ while(scanf("%d",&n)==1){ for(int i=1;i<=n;i++){head[i]=0;ans[i]=0;}cnt=0;sz=0; memset(isroot,true,sizeof(isroot)); int root,u,v,w; for(int i=1;i<=n;i++){ scanf("%d",&u); scanf(" :( %d ) ",&w); for(int j=1;j<=w;j++){ scanf("%d",&v); isroot[v]=false; ins(u,v); } } for(int i=1;i<=n;i++)if(isroot[i])root=i; dfs(root,0);rmq_init(); int q;scanf("%d",&q); while(q--){ scanf(" ( %d %d ) ",&u,&v); ans[LCA(u,v)]++; } for(int i=1;i<=n;i++) if(ans[i])printf("%d:%d\n",i,ans[i]); } return 0; }