LCA 13分 A1,6
查看原帖
LCA 13分 A1,6
373198
WhileTrueRP楼主2023/8/26 09:19
#include<iostream>
#include<vector>
using namespace std;
const int N = 5e4+5;
vector<int> a[N];
int d[N],anc[N][20],sum[N],ans;
int add(int u,int v){
	a[u].push_back(v);
	a[v].push_back(u);
}
void dfs(int u,int fa){
	for(int i=0;i<a[u].size();i++){
		int v = a[u][i];
		if(v == fa){
			continue;
		}
		d[v] = d[u] + 1;
		anc[v][0] = u;
		dfs(v,u);
	} 
}
void init(int n){
	for(int j=1;j<=18;j++){
		for(int i=1;i<=n;i++){
			anc[i][j] = anc[anc[i][j-1]][j-1];
		} 
	}
}
int lca(int u,int v){
	if(d[u] < d[v]){
		swap(u,v);
	}
	for(int i=18;i>=0;i--){
		if(d[anc[u][i]] >= d[v]){
			u = anc[u][i];
		}
	}
	if(u == v){
		return u;
	}
	for(int i=18;i>=0;i--){
		if(anc[u][i] != anc[v][i]){
			u = anc[u][i];
			v = anc[v][i];
		}
	}
	return anc[u][0];
}
void solve(int u,int fa){
	for(int i=0;i<a[u].size();i++){
		int v = a[u][i];
		if(v == fa){
			continue;
		}
		sum[v] += sum[u];
		ans = max(ans,sum[v]);
		solve(v,u);
	}
}
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v);
	}
	d[1] = 1;
	dfs(1,0);
	init(n);
	for(int i=1;i<=k;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		int l = lca(u,v);
		sum[l] ++;
		if(l != u){
			for(int i=0;i<a[u].size();i++){
				if(d[a[u][i]] > d[u]){
					sum[a[u][i]] --;
				}
			}
		}
		if(l != v){
			for(int i=0;i<a[v].size();i++){
				if(d[a[v][i]] > d[v]){
					sum[a[v][i]] --;
				}
			}
		}
		
	}
	ans = -1e9;
	solve(1,0);
	printf("%d",ans);
}
2023/8/26 09:19
加载中...