noip提高組樹上問題模板[lca,樹剖]
阿新 • • 發佈:2018-11-04
/*樹*/ //lca int lca(int x,int y){//dep[root]=1** if(dep[x]<dep[y])swap(x,y); for(int i=18;i>=0;i--) if(dep[fa[x][i]]>=dep[y]) x=fa[x][i]; if(x==y) return x; for(int i=18;i>=0;i--) if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i]; return fa[x][0]; } //樹鏈剖分 void dfs1(int u,int f){ size[u]=1; for(int i=first[u];i;i=next[i]){ int t=to[i];if(t==f)continue; dep[t]=dep[u]+1,fa[t]=u; dfs1(t,u); size[u]+=size[t]; if(size[t]>size[son[u]]) son[u]=t; } } void dfs2(int u,int top){ id[u]=++cnt , pre[cnt]=u , Top[u]=top; if(!son[u]) return; dfs(son[u],top); for(int i=first[u];i;i=next[i]){ int t=to[i];if(t==fa[u]||t==son[u]) continue; dfs(t,t); } } int Quary_val(int x,int y){ int ans=0; while(top[x]!=top[y]){ if(dep[top[x]]<dep[top[y]]) swap(x,y); ans += quary_val(1,1,n,id[top[x]],id[x]); x=fa[top[x]]; } if(dep[x]<dep[y]) swap(x,y); ans += quary_val(1,1,n,id[y],id[x]); return ans; }