P3128 [USACO15DEC]Max Flow P

题目

题目

思路

LCA+树上差分板子
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath> 
using namespace std;
int fa[500010][27],head[500010],c[500010],cf[500010],uo[500010],x,y;
struct f{
	int to,net;
} a[1000020];
int n,m,s;
void dfs(int x,int f)
{
	fa[x][0]=f;
	c[x]=c[f]+1;
	for (int j=1;(1<<j)<=c[x];j++)
	{
		fa[x][j]=fa[fa[x][j-1]][j-1];
	}
	for (int i=head[x];i!=-1;i=a[i].net)
	{
		if (a[i].to!=f) dfs(a[i].to,x);
	}
	return;
}
int k;
void add(int u,int v)
{
    a[k].to=v;
	a[k].net=head[u];
    head[u]=k++;
    return; 
}
int LCA(int x,int y)
{
	if (c[x]>c[y]) swap(x,y);
	for (int i=20;i>=0;i--)
	{
		if (c[x]<=c[fa[y][i]]) y=fa[y][i];
	}
	if (x==y) return x;
	for (int i=20;i>=0;i--)
	{
		if (fa[x][i]==fa[y][i]) continue;
		else x=fa[x][i],y=fa[y][i];
	}
	return fa[x][0];
}
void dfs1(int x,int f)
{
	for (int i=head[x];i!=-1;i=a[i].net)
	{
		if (a[i].to!=f)
		{
			dfs1(a[i].to,x);
			cf[x]+=cf[a[i].to];
		}
	}
	return;
}
int main()
{
    memset(head,-1,sizeof(head));
	cin>>n>>m;
	for (int i=1;i<n;i++)
	{
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	for (int i=1;i<=m;i++)
	{
		cin>>x>>y;
		int u=LCA(x,y);
		cf[x]++,cf[y]++,cf[u]--,cf[fa[u][0]]--;
	}
	dfs1(1,0);
	int ans=0;
	for (int i=1;i<=n;i++) ans=max(ans,cf[i]);
	cout<<ans;
    return 0;
}