P3128 [USACO15DEC] Max Flow P题解（树上差分，最近公共祖先，图论）

const int N = 5e4 + 5;
const int M = N - 1;

int cnt;
int head[N];
int fa[N][21];
int deepth[N];
int power[N];
int ans;

struct EDGE
{
	int v;
	int next;
}EDGE[M * 2];

void add(int u, int v)
{
	cnt++;
	EDGE[cnt].v = v;
	EDGE[cnt].next = head[u];
	head[u] = cnt;
}

void dfs(int son, int father)
{
	//第2^0个父亲就是这个父亲
	fa[son][0] = father;
	//儿子深度 = 父亲深度 + 1
	deepth[son] = deepth[father] + 1;
	//算低2 ^ i个父亲是谁
	for (int i = 1; (1 << i)/*注意不是i << 1*/ <= deepth[son]; i++)
		fa[son][i] = fa[fa[son][i - 1]][i -1];//公式
	for (int i = head[son]; i; i = EDGE[i].next)
	{
		int v = EDGE[i].v;
		if (v != father)/**/
			dfs(v, son);
	}
}

int lca(int x, int y)
{
	if (deepth[x] < deepth[y])//要让x在y下面，这样子方便后面统一处理
		swap(x, y);

	//使得x,y位于同一高度
	for (int i = 20; i >= 0; i--)//注意是逆序（原因:1、从上往下找比较快 2、若为顺序，则越往上走，找的父亲跨度越大，不符合要求）
	{
		if (deepth[fa[x][i]] >= deepth[y])
			x = fa[x][i];
	}
	if (x == y)//如果两个点已经重合
		return x;
	//找公共祖先且使得x,y位于公共祖先的下一层
	for (int i = 20; i >= 0; i--)
	{
		if (fa[x][i] != fa[y][i])
		{
			x = fa[x][i];
			y = fa[y][i];
		}
	}
	return fa[x][0];//因为位于公共祖先的下一层，所以他们的父亲就是公共祖先
}

void getans(int son, int father)
{
	for (int i = head[son]; i; i = EDGE[i].next)
	{
		if (EDGE[i].v == father)/**/
			continue;
		
		getans(EDGE[i].v, son);
		power[son] += power[EDGE[i].v];
	}
	ans = max(ans, power[son]);
}

void solve()
{
	int n, k;
	cin >> n >> k;
	int u, v, w;

	//建树
	for (int i = 0; i < n - 1; i++)
	{
		cin >> u >> v;
		add(u, v);
		add(v, u);
	}
	dfs(1, 0);//求第2 ^ n个父亲

	//求公共祖先、树上差分
	for (int i = 0; i < k; i++)
	{
		cin >> u >> v;
		int togetherfather = lca(u, v);
		power[u]++;
		power[v]++;
		power[togetherfather]--;
		power[fa[togetherfather][0]]--;
	}
	getans(1, 0);
	cout << ans << endl;
}

int main()
{
	solve();
	return 0;
}