P3128 [USACO15DEC]最大流 [树上差分]

传送门

点权差分 , 点加 点加 , lca减 , fa[lca]减

#include<bits/stdc++.h>
#define N 50050
using namespace std;
int first[N],next[N*2],to[N*2],tot;
int n,m,fa[N][20],dep[N],val[N],ans;
void add(int x,int y){
	next[++tot]=first[x],first[x]=tot,to[tot]=y;
}
void dfs(int u,int f){
	for(int i=1;i<=18;i++)
		fa[u][i] = fa[fa[u][i-1]][i-1];
	for(int i=first[u];i;i=next[i]){
		int t=to[i]; if(t==f) continue;
		fa[t][0]=u,dep[t]=dep[u]+1; dfs(t,u);
	}
}
int lca(int x,int y){
	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 dfs2(int u,int f){
	for(int i=first[u];i;i=next[i]){
		int t=to[i]; if(t==f) continue;
		dfs2(t,u); val[u] += val[t];
	} ans = max(ans,val[u]);
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++){
		int x,y; scanf("%d%d",&x,&y);
		add(x,y), add(y,x);
	} dep[1]=1,dfs(1,0);
	for(int i=1;i<=m;i++){
		int x,y; scanf("%d%d",&x,&y); int l=lca(x,y);
		val[x]++,val[y]++,val[l]--,val[fa[l][0]]--;
	} dfs2(1,0); printf("%d",ans); return 0;
}

猜你喜欢

转载自blog.csdn.net/sslz_fsy/article/details/85057645