BZOJ4719(NOIP2016)[天天爱跑步]--LCA+DFS

【链接】
bzoj4719

【解题报告】

对于一条在树上路径$S->T$我们可以将它拆为两条链即$S->LCA$和$LCA->T$。

所以我们只需要对两条链进行考虑。

不难得出$dep[S]-dep[x]=w[x]$从$S$出发刚好在$w[x]$时刻到达$x$。

定义$dis$为$S->T$的路径长度只有当$dep[T]-dep[x]=dis-w[x]$时从$T$出发刚好在$w[x]$时刻到达$x$。

$DFS$处理，用$ha$数组记录一下每个权值出现的次数，并用ans统计这个节点的答案。

最后需要注意从$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;
}