树链剖分求调
查看原帖
树链剖分求调
386424
文海郭向尚楼主2022/9/17 13:48

刚学树链剖分,提交全是RE,样例是对的,求调。

#include<bits/stdc++.h>
using namespace std;
const int N=500505;
int m,n,s;
vector <int> g[N];
struct node{
	int father;//父亲 
	int dep;//深度 
	int size;//子树上的节点个数(包括自己) 
	int son;//重儿子 
	int top;//重链顶部节点 
	int dfn;//dfs序 
}a[N];
int rnk[N];
int dfs1(int u){
	a[u].son=-1;
	a[u].size=1;
	int i,j;
	for(i=0;i<g[u].size();i++){
		int v=g[u][i];
		if(a[u].father!=v){
			a[v].dep=a[u].dep+1;
			a[v].father=u;
			dfs1(v);
			a[u].size=a[u].size+a[v].size;
			if(a[u].son==-1||a[v].size>a[a[u].son].size)a[u].son=v; 
		}
	}
}
int cnt=0;
void dfs2(int u,int t){
	a[u].top=t;
	cnt++;
	a[u].dfn=cnt;
	rnk[cnt]=u;
	if(a[u].son==-1)return ;
	dfs2(a[u].son,t);
	int i;
	for(i=0;i<g[u].size();i++){
		int v=g[u][i];
		if(v!=a[u].father&&v!=a[u].son){
			dfs2(v,v);
		}
	}
}
int lca(int u,int v){
	while(a[u].top!=a[v].top){
		if(a[a[u].top].dep>=a[a[v].top].dep){
			u=a[a[u].top].father;
		}else{
			v=a[a[v].top].father;
		}
	}
	if(a[u].dep<a[v].dep)return u;
	else return v;
}
int main(){
	cin>>n>>m>>s;
//	a[s].dep=1;
	int u,v; 
	for(int i=1;i<n;i++){
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	a[s].dep=0;
	dfs1(s);
	dfs2(s,s);
//	for(int i=1;i<=n;i++){
//		cout<<a[i].top<<" ";
//	} 
	for(int i=0;i<m;i++){
		cin>>u>>v;
		printf("%d\n",lca(u,v));
	}
} 
2022/9/17 13:48
加载中...