洛谷1600 天天爱跑步（树链剖分）

啊哈~

【题目分析】

哇这真的是DAY1T2吗。。。。。感觉要被强行劝退了。。。。。qwq

首先对于树上的路径s->t，我们都可以将其拆为s->lca(s,t)->t，我们用dona数组作为记录贡献的桶。

考虑一个点i能被观察到的情况，分为两种：

1.存在一条路径s->lca(s,t)，且depth[i]+val[i]（即原题的w）=depth[s]

2.存在一条路径lca(s,t)->t，且depth[i]-val[i]=depth[t]-lenth[s,t]（lenth数组记录s->t的路径长度）

所以我们考虑记录两个贡献：一个点在s->lca(s,t)上的贡献和lca(s,t)->t上的贡献，在dfs的时候我们记录dfs前和dfs后的差值，就是最后该点的贡献。（在代码中用1~MAXN维护第一个，MAXN+1~MAXN*2维护第二个）

对于树上的节点，维护三个信息：以该点为终点的路径编号，以该点为lca的路径的编号，以该点为起点的次数，用前两者统计对祖先的贡献，后者统计不会做出贡献的点，注意有可能lca会被重复计算，所以要在处理路径信息的时候提前减去。

反正写的很难受qwq，NOIP好难好难qwq

【代码~】

#include<bits/stdc++.h>
using namespace std;
const int MAXN=6e5+10;
const int MAXM=1e6+21000;

int n,m,cnt,s[MAXN],t[MAXN],lc[MAXN],lenth[MAXN];
int head[MAXN],depth[MAXN],fa[MAXN],son[MAXN],siz[MAXN],top[MAXN],val[MAXN],c[MAXN];
int nxt[MAXM],to[MAXM];
int start[MAXN],dona[MAXM<<1],a[MAXN],len[MAXN],ans[MAXN];
vector<int> anc[MAXN],end[MAXN];

int Read()
{
	int i=0,f=1;
	char c;
	for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
	if(c=='-')
	  f=-1,c=getchar();
	for(;c>='0'&&c<='9';c=getchar())
	  i=(i<<3)+(i<<1)+c-'0';
	return i*f;
}

void add(int x,int y)
{
	cnt++;
	nxt[cnt]=head[x];
	head[x]=cnt;
	to[cnt]=y;
}

void dfs1(int u,int fat)
{
	siz[u]=1,fa[u]=fat,depth[u]=depth[fat]+1;
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v!=fa[u])
		{
			c[v]=c[u]+1;
			dfs1(v,u);
			siz[u]+=siz[v];
			if(siz[son[u]]<siz[v])
			  son[u]=v;
		}
	}
}

void dfs2(int u,int tp)
{
	top[u]=tp;
	if(!son[u])
	  return ;
	dfs2(son[u],tp);
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v!=son[u]&&v!=fa[u])
		{
			dfs2(v,v);
		}
	}
}

int lca(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(depth[top[x]]<depth[top[y]])
		  swap(x,y);
		x=fa[top[x]];
	}
	return depth[x]<depth[y]?x:y;
}

void dfs3(int u)
{
    int xx=dona[depth[u]+val[u]],yy=dona[val[u]-depth[u]+MAXN]; 
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa[u]) 
		  continue;
        dfs3(v);
    }
    dona[depth[u]]+=start[u];
    for(int i=0;i<end[u].size();i++)
    {
        int v=end[u][i];
        dona[lenth[v]-depth[t[v]]+MAXN]++;
    }
    ans[u]+=dona[depth[u]+val[u]]-xx+dona[val[u]-depth[u]+MAXN]-yy;
    for(int i=0;i<anc[u].size();i++)
    {
        int v=anc[u][i];
        dona[depth[s[v]]]--,dona[lenth[v]-depth[t[v]]+MAXN]--;
    }
    return;
}

int main()
{
	n=Read(),m=Read();
	for(int i=1;i<n;++i)
	{
		int x=Read(),y=Read();
		add(x,y),add(y,x);
	}
	for(int i=1;i<=n;++i)
	  val[i]=Read();
	dfs1(1,0);
	dfs2(1,1);
	for(int i=1;i<=m;++i)
	{
		s[i]=Read(),t[i]=Read();
		lc[i]=lca(s[i],t[i]);
		lenth[i]=c[s[i]]+c[t[i]]-c[lc[i]]*2;
		anc[lc[i]].push_back(i);
		end[t[i]].push_back(i);
		start[s[i]]++;
		if(depth[s[i]]==depth[lc[i]]+val[lc[i]])
		  ans[lc[i]]--;
	}
	dfs3(1);
	for(int i=1;i<=n;++i)
	  cout<<ans[i]<<' ';
	return 0;
}