luogu P1600 天天爱跑步

题面传送门
这道题目蛮好的。
首先暴力不可取，会\(T\)
考虑分类讨论，把一段路径分为上行与下行。
设\(d_x\)为\(x\)节点的深度，则在上行时被观察到当且仅当\(d_v+t_v=d_u\)，下行也可推出当且仅当\(d_v-t_v=d_u\)
那么直接在树上差分哪一段有贡献。
求答案就是在子树内找与那个数相同的点的个数就好了。应该都会吧开桶记录即可。
代码实现:

#include<cstdio>
#include<cstring>
using namespace std;
int n,m,k,d[300039],lg[300039],fa[300039][20],ans[300039],g[300039],f[600039],lcas,x,y,z;
struct yyy{int to,z;};
struct ljb{
	int head,h[300039];
	yyy f[600039];
	inline void add(int x,int y){
		f[++head]=(yyy){y,h[x]};
		h[x]=head;
	}
}s,a,b;
inline void dfs1(int x,int last){
	d[x]=d[last]+1;
	fa[x][0]=last;
	for(int i=1;i<=lg[d[x]];i++) fa[x][i]=fa[fa[x][i-1]][i-1];
	int cur=s.h[x];
	yyy tmp;
	while(cur!=-1){
		tmp=s.f[cur];
		if(tmp.to!=last) dfs1(tmp.to,x);
		cur=tmp.z;
	}
}
inline void swap(int &x,int &y){x^=y,y^=x,x^=y;}
inline int lca(int x,int y){
	if(d[x]<d[y])swap(x,y);
	while(d[x]>d[y]) x=fa[x][lg[d[x]-d[y]]-1];
	if(x==y) return x;
	for(int i=lg[d[x]]-1;i>=0;i--) if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
inline void dfs2(int x){
	int ans1=f[d[x]+g[x]+n],ans2=f[d[x]-g[x]+n],cur=a.h[x];
	yyy tmp;
	while(cur!=-1){
		tmp=a.f[cur];
		f[tmp.to+n]++;
		cur=tmp.z;
	}
	cur=b.h[x];
	while(cur!=-1){
		tmp=b.f[cur];
		f[tmp.to+n]--;
		cur=tmp.z;
	}
	cur=s.h[x];
	while(cur!=-1){
		tmp=s.f[cur];
		if(tmp.to!=fa[x][0]) dfs2(tmp.to);
		cur=tmp.z;
	}
	//printf("%d %d %d %d %d\n",x,ans1,f[d[x]+g[x]+n],ans2,f[d[x]-g[x]+n]);
	if(!g[x]) ans[x]=f[d[x]+g[x]+n]-ans1;
	else ans[x]=f[d[x]+g[x]+n]-ans1+f[d[x]-g[x]+n]-ans2;
}
inline void read(int &x){
	char s=getchar();x=0;
	while(s<'0'||s>'9') s=getchar();
	while(s>='0'&&s<='9') x=(x<<3)+(x<<1)+(s^48),s=getchar();
}
inline void print(int x){
	if(x>9) print(x/10);
	putchar(x%10+48);
}
int main(){
	register int i;
	memset(b.h,-1,sizeof(b.h));
	memset(a.h,-1,sizeof(a.h));
	memset(s.h,-1,sizeof(s.h));
	read(n);read(m);
	for(i=1;i<=n;i++) lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	for(i=1;i<n;i++) read(x),read(y),s.add(x,y),s.add(y,x);
	dfs1(1,0);
	for(i=1;i<=n;i++) read(g[i]);
	for(i=1;i<=m;i++){
		read(x);read(y);
		lcas=lca(x,y);
		a.add(x,d[x]);
		b.add(fa[lcas][0],d[x]);
		a.add(y,d[lcas]*2-d[x]);
		b.add(lcas,d[lcas]*2-d[x]);
	}
	dfs2(1);
	for(i=1;i<=n;i++) print(ans[i]),putchar(' ');
}