洛谷 P1600 天天爱跑步【LCA】【桶】

---

题目：

传送门

---

题意：

在一棵树上，有$m$个人会从$0$秒开始由各自的起点跑向各自的终点，在每个点上都有一名观察员，他们会在给定的$w_i$时进行观察，当且仅当一个人在$w_i$时恰好到达该点才会有一的贡献，问每个观察员的答案是多少

---

分析：

一个人从起点到终点，我们不妨将路径一分为二，起点到$LCA$，$LCA$到终点
对于答案的计算，我们根据起点和终点来做一个桶，每个点保存一下什么时候会被贡献，到什么时候就结束
然后跑一边$dfs$，不要忘了将子树中对该点的贡献也一起计算

---

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<queue>
#include<map>
#define LL long long
using namespace std;
inline LL read()
{
	int s=0,f=1; char c=getchar();
	while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
	while(c>='0'&&c<='9') {s=s*10+c-'0';c=getchar();}
	return s*f;
}
struct node{
	int to,next;
}e[600005];
int ls[300005],cnt=0;
void add(int x,int y)
{
	e[cnt]=((node){y,ls[x]});
	ls[x]=cnt++;
	return;
}
int fa[300005][20],d[300005],w[300005],l;
void pdfs(int u)
{
	for(int i=ls[u];~i;i=e[i].next)
	{
		int v=e[i].to;
		if(d[v]) continue;
		d[v]=d[u]+1;
		fa[v][0]=u;
		pdfs(v);
	}
	return;
}
int LCA(int x,int y)
{
	if(d[x]>d[y]) swap(x,y);
	for(int i=l;i>=0;i--)
	  if(d[fa[y][i]]>=d[x]) y=fa[y][i];
	if(x==y) return x;
	for(int i=l;i>=0;i--) if(fa[x][i]!=fa[y][i])  x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int c1[300005],cup[600005],cdown[600005],ans[300005];
vector<int> c2[300005],eu[300005],ed[300005];
void dfs(int u,int f)
{
	int N=300000;
	int nup=d[u]+w[u]+N,ndown=d[u]-w[u]+N;
	int last=cup[nup]+cdown[ndown];
	cup[d[u]+N]+=c1[u];
	for(int i=0;i<c2[u].size();i++) cdown[c2[u][i]+N]++;
	for(int i=0;i<eu[u].size();i++) cup[eu[u][i]+N]--;
	for(int i=0;i<ed[u].size();i++) cdown[ed[u][i]+N]--;
	for(int i=ls[u];~i;i=e[i].next)
	{
		int v=e[i].to;
		if(v==f) continue;
		dfs(v,u);
	}
	ans[u]=cup[nup]+cdown[ndown]-last;
	return;
}
int main()
{
	memset(ls,-1,sizeof(ls));
	int n=read(),m=read();
	for(int i=1;i<n;i++)
	{
		int x=read(),y=read();
		add(x,y);add(y,x);
	}
	d[1]=1;
	pdfs(1);
	l=(int)(log(n)/log(2))+1;
	for(int j=1;j<=l;j++)
	  for(int i=1;i<=n;i++)
	    fa[i][j]=fa[fa[i][j-1]][j-1];
	for(int i=1;i<=n;i++) w[i]=read();
	for(int i=1;i<=m;i++)
	{
		int s=read(),t=read();
		int a=LCA(s,t);
		c1[s]++;c2[t].push_back(2*d[a]-d[s]);
		eu[fa[a][0]].push_back(d[s]);ed[a].push_back(2*d[a]-d[s]);
	}
	dfs(1,0);
	for(int i=1;i<=n;i++) printf("%d ",ans[i]);
	return 0;
}