洛谷 3128 树上差分

      这道题由于只是在最后询问，因此我们可以利用一种叫做差分的技巧，具体来说，就是如果要修改u到v的路径，我们可以将sum[u]++,sum[v]++,sum[lca(u,v)]--,sum[father[lca(u,v)]]--。这样我们可以通过一遍dfs自底向上转移sum值，这样就能最后在O(n)的时间内求出每个点的值。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 50005
int pre[maxn*2],other[maxn*2],last[maxn];
int n,m,l,jump[maxn][17],dep[maxn],ans[maxn],sum[maxn];
bool vis[maxn];

void connect(int x,int y) 
{
	l++;
	pre[l]=last[x];
	last[x]=l;
	other[l]=y;
}	

void pre_dfs(int u) 
{
	vis[u]=1;
	for (int p=last[u];p;p=pre[p]) 
	{
		int v=other[p];
		if (vis[v]) continue;
		jump[v][0]=u;
		dep[v]=dep[u]+1;
		pre_dfs(v);
	}
}

void dfs(int u) 
{
	vis[u]=1;
	ans[u]=sum[u];
	for (int p=last[u];p;p=pre[p]) 
	{
		int v=other[p];
		if (vis[v]) continue;
		dfs(v);
		ans[u]+=ans[v];
	}
}

int lca(int u,int v)
{
	if (dep[u]<dep[v]) swap(u,v);
	for (int j=0;j<=16;j++) 
		if ((dep[u]-dep[v])&(1<<j)) u=jump[u][j];
	if (u==v) return u;
	for (int j=16;j>=0;j--) 
		if (jump[u][j]!=jump[v][j]) 
		{
			u=jump[u][j];
			v=jump[v][j];	
		}
	return jump[u][0];
}

int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<n;i++) 
	{
		int a,b;
		scanf("%d%d",&a,&b);
		connect(a,b);connect(b,a);
	}
	pre_dfs(1);
	for (int j=1;j<=16;j++) 
		for (int i=1;i<=n;i++) jump[i][j]=jump[jump[i][j-1]][j-1];
	for (int i=1;i<=m;i++) 
	{
		int a,b;
		scanf("%d%d",&a,&b);
		int rt=lca(a,b);	
		sum[a]++;sum[b]++;sum[rt]--;
		sum[jump[rt][0]]--;
	}
	memset(vis,0,sizeof vis);
	dfs(1);
	int tt=0;
	for (int i=1;i<=n;i++) 
		tt=max(tt,ans[i]);
	printf("%d\n",tt);
	return 0;
}