比赛 csp2025模拟练习2 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 天天爱跑步 最终得分 100
用户昵称 李金泽 运行时间 3.683 s
代码语言 C++ 内存使用 51.10 MiB
提交时间 2025-10-29 10:43:30
显示代码纯文本
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
#define N 300005
#define fo(i,l,r) for(int i=l;i<=r;i++)
#define rf(i,r,l) for(int i=r;i>=l;i--)
#define ll long long
#define ul unsigned long long
#define int long long
using namespace std;
int n,m,h[N<<1],cnt,w[N],d[N],f[N][20],bs[N],bt[N*3],ans[N],x,y,z;const int eps=6e5;
struct edge{int v,n;}e[N<<1];
void ad(int u,int v){e[++cnt]={v,h[u]};h[u]=cnt;}
vector<int>s[N],t[N],l[N];
void dfs1(int u,int fa)
{
	d[u]=d[fa]+1;f[u][0]=fa;
	for(int i=h[u];i;i=e[i].n)
	{
		int v=e[i].v;
		if(v==fa)continue;
		dfs1(v,u);
	}
}
void init()
{
	fo(k,1,19)
		fo(i,1,n)
			f[i][k]=f[f[i][k-1]][k-1];
}
int lca(int x,int y)
{
	if(d[x]<d[y])swap(x,y);
	for(int i=19;~i;i--)
		if(d[f[x][i]]>=d[y])
			x=f[x][i];
	if(x==y)return x;
	for(int i=19;~i;i--)
		if(f[x][i]^f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
void dfs2(int u,int f)
{
	ans[u]-=bs[w[u]+d[u]]+bt[w[u]-d[u]+eps];
	for(int i=h[u];i;i=e[i].n)
		if(e[i].v^f)
			dfs2(e[i].v,u);
	int x;
	x=s[u].size();fo(i,0,x-1)bs[s[u][i]]++;
	x=t[u].size();fo(i,0,x-1)bt[t[u][i]+eps]++;
	x=l[u].size();fo(i,0,x-1)bs[l[u][i]]--;
	ans[u]+=bs[w[u]+d[u]]+bt[w[u]-d[u]+eps];
	x=l[u].size();fo(i,0,x-1)bt[l[u][i]-2*d[u]+eps]--;
}
void swap(int &x,int &y){int t=x;x=y;y=t;}
int max(int x,int y){return x<y?x:y;}
int min(int x,int y){return x<y?x:y;}
int read(){
	int sum=0;bool f=0;char c=getchar();
	for(;c<48||c>57;c=getchar())if(c==45)f=1;
	for(;c>=48&&c<=57;c=getchar())sum=sum*10+(c&15);
	return f?-sum:sum;
}
signed main(){
	freopen("runninga.in","r",stdin);freopen("runninga.out","w",stdout);
	n=read();m=read();
	fo(i,1,n-1)x=read(),y=read(),ad(x,y),ad(y,x);
	fo(i,1,n)w[i]=read();
	dfs1(1,0);init();
	while(m--)
	{
		x=read();y=read();z=lca(x,y);
		s[x].push_back(d[x]);t[y].push_back(d[x]-2*d[z]);
		l[z].push_back(d[x]);
	}
	dfs2(1,0);
	fo(i,1,n)printf("%lld ",ans[i]);
	return 0;
}