「USACO2015」 最大流 - 树上差分

题目大意

给定一棵有N个点的树，所有节点的权值都为0。有K次操作，每次指定两个点s,t，将s到t路径上所有点的权值都加一，最后输出K次操作完毕后权值最大的那个点的权值。

分析

算得上是树上差分的模板题了。

说一下普通的差分。现在有这么一个问题，给定一个序列A，有K个修改，每个修改将[L,R]中的数加1，最后问其中的最大数。最普通的做法就是每次跑一遍[L,R]，并更新最大值，显然这样做可能会TLE；而差分的做法就将区间修改转化为点修改。定义差分序列D，，易知D的前缀和就是原数组。而如果在处+1，则对于所有的都会+1，在处-1，则对于所有的都会-1，重叠部分与+1抵消，故改变的只有[L,R]。修改结束后，统计一遍前缀和，找最大数即可解决。

同理在树上的差分也是如此。若想让图中u到v上的每条边的权值+1，可以让，其中表示号节点与它父亲之间的边，最后通过一次DFS，统计前缀和，即（i是j的父亲）。

对于点权的差分，大体是差不多的，减的时候让，因为对于点，在这条链上，而它只能加一个，所以++，它++，它的父亲也会++，所以它的父亲要--。

 本题根据题意是点的差分，所以按照第二种方法即可。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <map>
#include <queue>
using namespace std;
struct node {
	int to,next;
}e[50005*2];
int h[50005],cnt,dth;
int n,k,dep[50005],ans;
int f[50005][20],v[50005];
void add(int x,int y) {
	e[++cnt]=(node){y,h[x]};
	h[x]=cnt;
}
void FindDepth(int x,int prt) {
	dep[x]=dep[prt]+1;
	for (int i=h[x];i;i=e[i].next) {
		int y=e[i].to;
		if (y==prt) continue;
		f[y][0]=x;
		FindDepth(y,x);
	}
}
int GetLca(int x,int y) {
	if (dep[x]<dep[y]) swap(x,y);
	for (int i=dth;i>=0;i--)
		if (dep[f[x][i]]>=dep[y]) x=f[x][i];
	if (x==y) return x;
	for (int i=dth;i>=0;i--)
		if (f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
void Dfs(int x,int prt) {
	for (int i=h[x];i;i=e[i].next) {
		int y=e[i].to;
		if (y==prt) continue;
		Dfs(y,x);
		v[x]+=v[y];
	}
	ans=max(ans,v[x]);
}
int main() {
	scanf("%d%d",&n,&k);
	for (int i=1,a,b;i<n;i++) {
		scanf("%d%d",&a,&b);
		add(a,b);
		add(b,a);
	}
	dth=(int)(log(n)/log(2));
	FindDepth(1,0);
	for (int j=1;(1<<j)<=n;j++)
		for (int i=1;i<=n;i++)
			f[i][j]=f[f[i][j-1]][j-1];
	
	for (int i=1,a,b;i<=k;i++) {
		scanf("%d%d",&a,&b);
		v[a]++;
		v[b]++;
		int t=GetLca(a,b);
		v[t]--;
		v[f[t][0]]--;
	}
	Dfs(1,0);
	printf("%d",ans);
	return 0;
}