【NOIP2016】天天爱跑步

天天爱跑步

参考了https://www.cnblogs.com/ljh2000-jump/p/6189053.html的博客

详细做法都在注释里

 

 

//https://www.cnblogs.com/ljh2000-jump/p/6189053.html
#include<bits/stdc++.h>
using namespace std;

const int maxn=300000+5;
const int mov=maxn;

int n,m;
int w[maxn];

int ans[maxn],val[maxn],buck[maxn],buck1[maxn<<2];

int s[maxn],t[maxn],L[maxn],len[maxn],md;
int depth[maxn];
vector<int>g[maxn];
vector<int>v1[maxn],v2[maxn],v3[maxn];
int anc[maxn][20];
void addedge(int a,int b)
{
	g[a].push_back(b);
	g[b].push_back(a);
}

//------------------------lca
void rec(int x)
{
	for(int j=1; j<=19; j++)
	{
		if((1<<j)>=depth[x])break;
		anc[x][j]=anc[anc[x][j-1]][j-1];
	}
	for(int i=0; i<(int)g[x].size(); i++)
	{
		if(depth[g[x][i]])continue;
		anc[g[x][i]][0]=x;
		depth[g[x][i]]=depth[x]+1;
		rec(g[x][i]);
	}
}
int lca(int a,int b)
{
	if(depth[a]<depth[b])swap(a,b);
	int dif=depth[a]-depth[b];
	for(int x=1,ti=0; x<=dif; x<<=1,ti++)
	{
		if(x&dif)
		{
			a=anc[a][ti];
		}
	}
	if(a==b)return a;
	for(int i=19; i>=0; i--)
	{
		if(anc[a][i]==anc[b][i])continue;
		a=anc[a][i];
		b=anc[b][i];
	}
	return anc[a][0];
}
void reset(int st)
{
	anc[st][0]=st;
	depth[st]=1;
	rec(st);
}
//-----------

void dfs1(int x,int f)//处理u到lca这一段 
{	//w[x]=dep[u]-dep[x] -> dep[u]=dep[x]+w[x]
	//buck 统计 dep[u] 个数 
	int nw=depth[x]+w[x],old=buck[nw];//差分 
	for(int i=0;i<(int)g[x].size();i++)//处理子树 
	{
		int to=g[x][i];
		if(to==f)continue;
		dfs1(to,x);
	}
	buck[depth[x]]+=val[x];//当前深度(dep[u])增加 从x出发的点的个数 
	ans[x]+=buck[nw]-old;//更新答案 
	for(int i=0;i<(int)v1[x].size();i++)//到lca就不会往上走了 删除 
	{
		buck[depth[v1[x][i]]]--;//lca to s[i]
	}
}

void dfs2(int x,int f)//处理lca到v这一段 
{	//len(u,v)-(dep[v]-dep[x])=w[x] ->w[x]-dep[x]=len(u,v)-dep[v]
	//-> dep[x]-w[x]=dep[v]-len(u,v)
	int nw=depth[x]-w[x]+mov,old=buck1[nw];//防止出负数 差分 
	for(int i=0;i<(int)g[x].size();i++)//处理子树 
	{
		int to=g[x][i];
		if(to==f)continue;
		dfs2(to,x);
	}
	for(int i=0;i<(int)v2[x].size();i++)
	{
		buck1[mov+v2[x][i]]++;//当前点(v) dep[x]-len(u,v)增加 从当前点向上走 
	}
	ans[x]+=buck1[nw]-old;//更新答案
	for(int i=0;i<(int)v3[x].size();i++)
	{
		buck1[mov+v3[x][i]]--;//到lca截止 删除 
	}
}

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin>>n>>m;
	for(int i=1; i<n; i++)
	{
		int x,y;
		cin>>x>>y;
		addedge(x,y);
	}
	reset(1);//lca
	
	for(int i=1; i<=n; i++)
	{
		cin>>w[i];
		md=max(md,depth[i]);
	}
	for(int i=1; i<=m; i++)
	{
		cin>>s[i]>>t[i];
		val[s[i]]++;
		L[i]=lca(s[i],t[i]);len[i]=depth[s[i]]-depth[L[i]]+depth[t[i]]-depth[L[i]];
		v1[L[i]].push_back(s[i]);
	}
	dfs1(1,0);
	for(int i=1;i<=m;i++)
	{
		//w[x]-dep[x]=len(u,v)-dep[v] -> dep[v]-len(u,v)=dep[x]-w[x]
		v2[t[i]].push_back(depth[t[i]]-len[i]);
		v3[L[i]].push_back(depth[t[i]]-len[i]);
	}
	dfs2(1,0);
	for(int i=1;i<=m;i++)
	{
		if(depth[s[i]]-depth[L[i]]==w[L[i]])ans[L[i]]--;//lca有贡献 上去下来多算了一次 
	}
	for(int i=1;i<=n;i++)printf("%d ",ans[i]);
	return 0;
}