「USACO2015」 最大流 - 树上差分

题目大意

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

分析

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

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

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

对于点权的差分,大体是差不多的,减的时候让w[LCA(u,v)]--,w[father[LCA(u,v)]]--,因为对于点,LCA(u,v)在这条链上,而它只能加一个,所以++,它++,它的父亲也会++,所以它的父亲要--。

「USACO2015」 最大流 - 树上差分_第1张图片

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

代码

#include 
#include 
#include 
#include 
#include 
#include 
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]=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

 

你可能感兴趣的:(图论---最近公共祖先LCA,图论---树上差分)