1. 程式人生 > >noip提高組樹上問題模板[lca,樹剖]

noip提高組樹上問題模板[lca,樹剖]

/*樹*/
//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;
}