【題目分析】
分析一下每個人的路徑,在樹中可以分爲上升的一段和下降的一段。那麼對於上升的一段,它的深度+觀察時間是一個定值的時候,纔可能被觀察到;對於下降的情況,它的觀察時間減去深度是一個定值。所以我們統計每個點他可以觀察到的值(分爲上升和下降兩部分進行討論)。然後只需要樹鏈剖分,打上nlogn個標記(可以用vector)。然後差分數組從前到後、跑一邊就可以了。
【代碼】(考場程序)
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int read()
{
int ret=0,f=1; char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){ret*=10;ret+=ch-'0';ch=getchar();}
return ret*f;
}
int n,m;
int h[300001],to[600001],ne[600001],en=0;
void add(int a,int b)
{
to[en]=b;
ne[en]=h[a];
h[a]=en++;
}
int root=1,dep[300001],fa[300001],son[300001],siz[300001];
vector <int> st[300001];
vector <int> ed[300001];
void dfs1(int k)
{
siz[k]=1;
for (int i=h[k];i>=0;i=ne[i])
if (to[i]!=fa[k]){
fa[to[i]]=k;
dep[to[i]]=dep[k]+1;
dfs1(to[i]);
siz[k]+=siz[to[i]];
if (siz[son[k]]<siz[to[i]]) son[k]=to[i];
}
}
int pos[300001],top[300001],li[300001],cnt=0;
void dfs2(int k,int tp)
{
top[k]=tp;
cnt++; pos[k]=cnt; li[cnt]=k;
if (!son[k]) return ;
if (son[k]) dfs2(son[k],tp);
for (int i=h[k];i>=0;i=ne[i])
if (to[i]!=son[k]&&to[i]!=fa[k]) dfs2(to[i],to[i]);
return ;
}
int w[300001];
vector <int> que[300001];
vector <int> bel[300001];
struct node{int s,t,lca;}q[300001];
int vis[300001],f[300001];
int gf(int k)
{
if (f[k]==k) return k;
else return f[k]=gf(f[k]);
}
void dfs3(int k)
{
f[k]=k;
vis[k]=1;
for (int i=0;i<que[k].size();++i)
if (vis[que[k][i]]) q[bel[k][i]].lca=gf(que[k][i]);
for (int i=h[k];i>=0;i=ne[i])
if (fa[k]!=to[i]){
dfs3(to[i]);
f[to[i]]=k;
}
return ;
}
void lca(){dfs3(1);}
int s[300001],ans[300001];
void add(int a,int b,int fn)
{
if (dep[a]<dep[b]) swap(a,b);
while (top[a]!=top[b])
{
if (dep[top[a]]<dep[top[b]]) swap(a,b);
st[pos[top[a]]].push_back(fn);
ed[pos[a]+1].push_back(fn);
a=fa[top[a]];
}
if (dep[a]<dep[b]) swap(a,b);
st[pos[b]].push_back(fn);
ed[pos[a]+1].push_back(fn);
}
void cal1()
{
for (int i=1;i<=n;++i) w[i]-=dep[i];
for (int i=1;i<=m;++i)
{
int at;
at=dep[q[i].s]-dep[q[i].lca];
if (w[q[i].lca]==at-dep[q[i].lca]) ans[q[i].lca]--;
add(q[i].lca,q[i].t,at-dep[q[i].lca]);
}
for (int i=1;i<=n;++i)
{
for (int j=0;j<st[i].size();++j) s[st[i][j]]++;
for (int j=0;j<ed[i].size();++j) s[ed[i][j]]--;
ans[li[i]]+=s[w[li[i]]];
}
}
void cal2()
{
for (int i=0;i<=n;++i) st[i].clear(),ed[i].clear();
memset(s,0,sizeof s);
for (int i=1;i<=n;++i) w[i]+=2*dep[i];
for (int i=1;i<=m;++i)
{
int at;
at=dep[q[i].s]-dep[q[i].lca];
add(q[i].lca,q[i].s,at+dep[q[i].lca]);
}
for (int i=1;i<=n;++i)
{
for (int j=0;j<st[i].size();++j) s[st[i][j]]++;
for (int j=0;j<ed[i].size();++j) s[ed[i][j]]--;
ans[li[i]]+=s[w[li[i]]];
}
}
void out()
{
for (int i=1;i<=n;++i)
printf("%d%c",ans[i],i==n?'\n':' ');
}
int main()
{
memset(h,-1,sizeof h);
n=read(); m=read();
for (int i=1;i<n;++i)
{
int a,b;
a=read(); b=read();
add(a,b); add(b,a);
}
for (int i=1;i<=n;++i)
w[i]=read();
for (int i=1;i<=m;++i)
{
int a,b;
a=read(); b=read();
que[a].push_back(b);
que[b].push_back(a);
bel[a].push_back(i);
bel[b].push_back(i);
q[i].s=a;q[i].t=b;
}
dfs1(root);
dfs2(root,root);
lca();
cal1();
cal2();
out();
fclose(stdin); fclose(stdout);
return 0;
}