【链接】
bzoj4719
【解题报告】
对于一条在树上路径 S−>T 我们可以将它拆为两条链即 S−>LCA 和 LCA−>T 。
所以我们只需要对两条链进行考虑。
不难得出
dep[S]−dep[x]=w[x]
从
S
出发刚好在
定义
最后需要注意从 S−>LCA 和 LCA−>T 两条链中 LCA 被处理了两次,最后特判一下抠去答案就行了。
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=300005,maxm=600005,maxv=900005,maxt=19,tt=299998;
int n,m,tot,lgn,dep[maxn],ans[maxn],st[maxn],gl[maxn],lca[maxn],f[maxn][maxt],w[maxn],ha[maxv],lnk[maxn],son[maxm],nxt[maxm];
struct Edge
{
int E,lnk[maxn],son[maxm],nxt[maxm];
void Add(int x,int y) {son[++E]=y; nxt[E]=lnk[x]; lnk[x]=E;}
}A,B,C,D;
inline char nc()
{
static char buf[100000],*l,*r;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
if (l==r) return EOF; return *l++;
}
inline int Read()
{
int res=0; char ch=nc();
while (ch<'0'||ch>'9') ch=nc();
while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=nc();
return res;
}
void Add(int x,int y) {son[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;}
void Dfs(int x,int fa)
{
for (int j=lnk[x]; j; j=nxt[j])
if (son[j]!=fa) f[son[j]][0]=x,dep[son[j]]=dep[x]+1,Dfs(son[j],x);
}
void BZ()
{
for (int j=1; j<=lgn; j++)
for (int i=1; i<=n; i++)
f[i][j]=f[f[i][j-1]][j-1];
}
int LCA(int x,int y)
{
if (dep[x]>dep[y]) swap(x,y);
for (int i=lgn; i>=0; i--)
if (dep[y]-(1<<i)>=dep[x]) y=f[y][i];
if (x==y) return x;
for (int i=lgn; i>=0; i--)
if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
return f[x][0];
}
int Getr(int x) {return x+tt;}
void Count1(int x,int fa)
{
int lst=ha[Getr(dep[x]+w[x])];
for (int j=lnk[x]; j; j=nxt[j])
if (son[j]!=fa) Count1(son[j],x);
for (int j=A.lnk[x]; j; j=A.nxt[j]) ha[Getr(A.son[j])]++;
ans[x]+=ha[Getr(dep[x]+w[x])]-lst;
for (int j=C.lnk[x]; j; j=C.nxt[j]) ha[Getr(C.son[j])]--;
}
void Count2(int x,int fa)
{
int lst=ha[Getr(dep[x]-w[x])];
for (int j=lnk[x]; j; j=nxt[j])
if (son[j]!=fa) Count2(son[j],x);
for (int j=B.lnk[x]; j; j=B.nxt[j]) ha[Getr(B.son[j])]++;
ans[x]+=ha[Getr(dep[x]-w[x])]-lst;
for (int j=D.lnk[x]; j; j=D.nxt[j]) ha[Getr(D.son[j])]--;
}
int main()
{
freopen("4719.in","r",stdin);
freopen("4719.out","w",stdout);
n=Read(); m=Read(); tot=A.E=B.E=C.E=D.E=0; lgn=log2(n);
memset(lnk,0,sizeof(lnk));
for (int i=1,x,y; i<n; i++) x=Read(),y=Read(),Add(x,y),Add(y,x);
for (int i=1; i<=n; i++) w[i]=Read();
dep[1]=1; Dfs(1,0); BZ();
for (int i=1; i<=m; i++)
{
st[i]=Read(); gl[i]=Read(); lca[i]=LCA(st[i],gl[i]); int dis=dep[st[i]]+dep[gl[i]]-2*dep[lca[i]];
A.Add(st[i],dep[st[i]]); B.Add(gl[i],dep[gl[i]]-dis);
C.Add(lca[i],dep[st[i]]); D.Add(lca[i],dep[gl[i]]-dis);
}
Count1(1,0); Count2(1,0);
for (int i=1; i<=m; i++) if (dep[st[i]]-dep[lca[i]]==w[lca[i]]) ans[lca[i]]--;
for (int i=1; i<n; i++) printf("%d ",ans[i]); printf("%d",ans[n]);
return 0;
}