树上差分的两种操作

       近几年来，树上差分也考地多了，这篇文章中讲一讲树上差分的两种基本操作。

一、已知路径求被所有路径覆盖的边

       首先对已知的这n条路径的起点a和终点b的权值+1，并对lca(a,b)的权值-2。

       从根节点开始深搜，回溯时将其本身的权值加上所有子节点的权值。

       那么满足要求的边就是权值等于n的节点与其父节点所连的边。

Code：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#define N 200005
using namespace std;
int head[N],a,b,tot,fa[N][21],f[N],deep[N],value[N];//value[i]表示i与i的父节点的连边被经过几次
inline int read()
{
	int x=0,f=1;char s=getchar();
	while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
	while(s<='9'&&s>='0'){x=x*10+s-'0';s=getchar();}
	return x*f;
}
struct edge
{
	int vet,next;
}edge[N];
void add(int u,int v)
{
	edge[++tot].vet=v;
	edge[tot].next=head[u];
	head[u]=tot;
}
void make(int u)
{
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].vet;
		if(v==fa[u][0])continue;
		deep[v]=deep[u]+1;
		fa[v][0]=u;
		make(v);
	}
}
void dfs(int u)
{
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].vet;
		if(v==fa[u][0])continue;
		dfs(v);
		value[u]+=value[v];
	}
}
int lca(int a,int b)
{
	if(deep[a]>deep[b])swap(a,b);
	int d=deep[b]-deep[a];
	for(int i=0;i<=20;i++)
		if((1<<i)&d)b=fa[b][i];
	if(a==b)return a;
	for(int i=20;i>=0;i--)
		if(fa[a][i]!=fa[b][i])
		{
			a=fa[a][i];
			b=fa[b][i];
		}
	return fa[a][0];
}
int main()
{
	memset(head,-1,sizeof(head));
	memset(f,0,sizeof(f));
	memset(value,0,sizeof(value));
	tot=0;
	int n=read(),k=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		add(u,v);add(v,u);
	}
	make(1);
	for(int j=1;j<=20;j++)
		for(int i=1;i<=n;i++)
			fa[i][j]=fa[fa[i][j-1]][j-1];
	for(int i=1;i<=k;i++)
	{
		int a=read(),b=read();
		value[a]++;value[b]++;
		value[lca(a,b)]-=2;
	}
	dfs(1);
	for(int i=2;i<=n;i++)
		if(value[i]==k)printf("%d %d\n",i,fa[i][0]);
	return 0;
}

二、已知路径求树上所有节点被路径覆盖次数

       对每条路径的起点a和终点b的权值+1，对lca(a,b)的权值-1，对lca(a,b)的父节点权值-1。

       从根节点开始深搜，回溯时将其本身的权值加上所有子节点的权值。

       每个节点的权值既是其被路径覆盖的次数。

       例题：USACO 2015 Dec Platinum。

Code：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#define N 200005
using namespace std;
int head[N],a,b,tot,fa[N][21],f[N],deep[N],value[N];
inline int read()
{
	int x=0,f=1;char s=getchar();
	while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
	while(s<='9'&&s>='0'){x=x*10+s-'0';s=getchar();}
	return x*f;
}
struct edge
{
	int vet,next;
}edge[N];
void add(int u,int v)
{
	edge[++tot].vet=v;
	edge[tot].next=head[u];
	head[u]=tot;
}
void make(int u)
{
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].vet;
		if(v==fa[u][0])continue;
		deep[v]=deep[u]+1;
		fa[v][0]=u;
		make(v);
	}
}
void dfs(int u)
{
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].vet;
		if(v==fa[u][0])continue;
		dfs(v);
		value[u]+=value[v];
	}
}
int lca(int a,int b)
{
	if(deep[a]>deep[b])swap(a,b);
	int d=deep[b]-deep[a];
	for(int i=0;i<=20;i++)
		if((1<<i)&d)b=fa[b][i];
	if(a==b)return a;
	for(int i=20;i>=0;i--)
		if(fa[a][i]!=fa[b][i])
		{
			a=fa[a][i];
			b=fa[b][i];
		}
	return fa[a][0];
}
int main()
{
	memset(head,-1,sizeof(head));
	memset(f,0,sizeof(f));
	memset(value,0,sizeof(value));
	tot=0;
	int n=read(),k=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		add(u,v);add(v,u);
	}
	make(1);
	for(int j=1;j<=20;j++)
		for(int i=1;i<=n;i++)
			fa[i][j]=fa[fa[i][j-1]][j-1];
	for(int i=1;i<=k;i++)
	{
		int a=read(),b=read();
		value[a]++;value[b]++;
		int Lca=lca(a,b);
		value[Lca]--;
		if(fa[Lca][0]>0)value[fa[Lca][0]]--;
	}
	dfs(1);
	int ans=0;
	for(int i=1;i<=n;i++)
		ans=max(ans,value[i]);
	printf("%d",ans);
	return 0;
}