BZOJ 4390: [Usaco2015 dec]Max Flow|树链剖分

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<algorithm>
#include<iostream>
#define O 50555
using namespace std;
int sc()
{
	int i=0; char c=getchar();
	while(c>'9'||c<'0')c=getchar();
	while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
	return i;
}
int head[O],lst[2*O],nxt[2*O],fa[O],top[O];
int size[O],ws[O],v[O],deep[O];
int n,m,tot,ans=0;
void insert(int x,int y)
{
	lst[++tot]=y;
	nxt[tot]=head[x];
	head[x]=tot;
}
void dfs1(int x,int f)
{
	size[x]=1,deep[x]=deep[f]+1;
	for(int i=head[x];i;i=nxt[i])
	    if(lst[i]!=f)
	    {
	    	fa[lst[i]]=x;
	    	size[x]++;
	    	dfs1(lst[i],x);
	    }
}
void dfs2(int x,int tp)
{
	int k=0; top[x]=tp;
	for(int i=head[x];i;i=nxt[i])
	    if(lst[i]!=fa[x]&&size[lst[i]]>size[k])
		    k=lst[i];
	if(k)dfs2(k,tp);
	for(int i=head[x];i;i=nxt[i])
	    if(lst[i]!=fa[x]&&lst[i]!=k)
		    dfs2(lst[i],lst[i]);
}
int Lca(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]])swap(x,y);
		x=fa[top[x]];
	}
	return deep[x]<deep[y]?x:y;
}
void dfs(int x)
{
	for(int i=head[x];i;i=nxt[i])
	    if(lst[i]!=fa[x])
	    {
	    	dfs(lst[i]);
	    	v[x]+=v[lst[i]];
	    }
	if(v[x]>ans)ans=v[x];
}
int main()
{
	n=sc();m=sc();
	for(int i=1;i<n;i++)
	{
		int x=sc(),y=sc();
		insert(x,y);
		insert(y,x);
	}
	dfs1(1,0);
	dfs2(1,1);
	for(int i=1;i<=m;i++)
	{
		int x=sc(),y=sc(),p=Lca(x,y);
		v[fa[p]]-=1,v[p]-=1;
		v[x]+=1,v[y]+=1;
	}
	dfs(1);
	cout << ans;
	return 0;
}