萌新初学树上差分 求助各位大佬！！！

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
struct node{
	int to,next;
}edge[1000001];
int head[1000001],cnt,m,n,s,depth[1000001],grand[1000001][21],N,f[100001],tmp[1000001];
void addedge(int u,int v)
{
	edge[++cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt;
}
void dfs(int u,int fa)
{
	depth[u]=depth[fa]+1;//
	for(int i=1;(1<<i)<=depth[u];i++)
	{
		grand[u][i]=grand[grand[u][i-1]][i-1];
	}
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=fa)
		{
			grand[v][0]=u;
			dfs(v,u);
		}
	}
}
void Int()
{
	N=log(n);
	depth[f[1]]=0;	
	dfs(f[1],0);
}
int lca(int a,int b)
{
	if(depth[a]<depth[b]) swap(a,b);
	for(int i=N;i>=0;i--)
	{
		if(depth[grand[a][i]]>=depth[b])//
		{
			a=grand[a][i];	
		}
	}
	if(a==b) return a;
	for(int i=N;i>=0;i--)
	{
		if(grand[a][i]!=grand[b][i])
		{
			a=grand[a][i];
			b=grand[b][i];
		}
	}
	return grand[a][0];
}
void dfs1(int u,int fa)
{
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=fa)
		{
			dfs1(v,u);
			tmp[u]+=tmp[v];
		}
	}
}
int main()
{
	scanf("%d",&n);     //scanf("%d%d%d",&n,&m,&s);
	for(int i=1;i<=n;i++) scanf("%d",&f[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		addedge(u,v);
		addedge(v,u);
	}
	Int();
	for(int i=1;i<n;i++)
	{
		tmp[f[i]]++,tmp[f[i+1]]++;
		tmp[lca(f[i],f[i+1])]--,tmp[grand[lca(f[i],f[i+1])][0]]--;
	}
	dfs1(f[1],0);
	for(int i=2;i<=n;i++) tmp[f[i]]--;
	for(int i=1;i<=n;i++)
	{
		printf("%d\n",tmp[i]);
	}
}