洛谷 3128 树上差分

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

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