[BZOJ4719][NOIP2016]天天爱跑步-LCA+树上差分

天天爱跑步

Description

小c同学认为跑步非常有趣,于是决定制作一款叫做《天天爱跑步》的游戏。?天天爱跑步?是一个养成类游戏,需要玩家每天按时上线,完成打卡任务。

这个游戏的地图可以看作一一棵包含 N个结点和N-1 条边的树, 每条边连接两个结点,且任意两个结点存在一条路径互相可达。树上结点编号为从1到N的连续正整数。

现在有个玩家,第个玩家的起点为Si ,终点为Ti 。每天打卡任务开始时,所有玩家在第0秒同时从自己的起点出发, 以每秒跑一条边的速度,不间断地沿着最短路径向着自己的终点跑去, 跑到终点后该玩家就算完成了打卡任务。 (由于地图是一棵树, 所以每个人的路径是唯一的)

小C想知道游戏的活跃度, 所以在每个结点上都放置了一个观察员。 在结点的观察员会选择在第Wj秒观察玩家, 一个玩家能被这个观察员观察到当且仅当该玩家在第Wj秒也理到达了结点J 。 小C想知道每个观察员会观察到多少人?

注意: 我们认为一个玩家到达自己的终点后该玩家就会结束游戏, 他不能等待一 段时间后再被观察员观察到。 即对于把结点J作为终点的玩家: 若他在第Wj秒重到达终点,则在结点J的观察员不能观察到该玩家;若他正好在第Wj秒到达终点,则在结点的观察员可以观察到这个玩家。

Input

第一行有两个整数N和M 。其中N代表树的结点数量, 同时也是观察员的数量, M代表玩家的数量。
接下来n-1 行每行两个整数U和V ,表示结点U 到结点V 有一条边。
接下来一行N 个整数,其中第个整数为Wj , 表示结点出现观察员的时间。
接下来 M行,每行两个整数Si和Ti,表示一个玩家的起点和终点。
对于所有的数据,保证 。
1<=Si,Ti<=N,0<=Wj<=N

Output

输出1行N 个整数,第个整数表示结点的观察员可以观察到多少人。

Sample Input

6 3
2 3
1 2
1 4
4 5
4 6
0 2 5 1 2 3
1 5
1 3
2 6

Sample Output

1 2 1 0 1

HINT

对于1号点,Wi=0，故只有起点为1号点的玩家才会被观察到，所以玩家1和玩家2被观察到，共有2人被观察到。
对于2号点，没有玩家在第2秒时在此结点，共0人被观察到。
对于3号点，没有玩家在第5秒时在此结点，共0人被观察到。
对于4号点，玩家1被观察到，共1人被观察到。
对于5号点，玩家1被观察到，共1人被观察到。
对于6号点，玩家3被观察到，共1人被观察到。

---

很久很久以前。有一个萌新。
在考场上遇到这题时，它单纯到求LCA只会暴力和树链剖分，不知道树上差分是什么，甚至连面向数据编程的重要思想都不知道。
走出NOIP2016的考场后，它才知道Day1T2可以面向数据骗80分，然而它只打了一个树链剖分找LCA的模拟，得了25分。
这个萌新就是咱……

(╯‵□′)╯︵┻━┻

---

思路:
NOIP竟然会出树上差分……咱这种蒟蒻最近才听说这种方法……
所以，首先连咱这种蒟蒻都想到了拆路径的方法，也就是把每个玩家的路径拆成一条到LCA的路径和从LCA到终点的路径~
然后，使用树上差分统计答案即可~

那么，树上差分是什么?
差分的具体思想是，当某区间内某元素对答案有贡献，就在区间起点打一个+1标记代表多出了一个对答案有贡献的元素，在终点打一个-1标记代表一个对答案有贡献的元素在该位置结束了它的使命。
于是，统计答案就变成了维护扫到当前位置为止的标记个数，直接累加就是答案~

首先对从下到上的路径观察~
然后就会发现:所有对某点上的观察员i有贡献的点只可能是以i点往下w[i]层深度的任一后代为起点的玩家~
然后我们就可以维护一个标记数组，每个位置分别表示当前共访问了多少个深度为当前位置下标的点。

直接dfs,先对目标深度，也就是对当前观察员i有贡献的深度为dep[i]+w[i]的标记，记录其当前的值。

然后递归搜索所有子树，每搜到一个点便把以当前点为起点的所有路径加入当前点深度的标记中，代表这些点开始产生贡献，并把以当前点为LCA的所有路径从从该路径起点所在深度中减去，代表这些点停止产生贡献。

当回溯到当前点，统计搜索完所有子树后dep[i]+w[i]处标记的新值，并减去记录下的旧的值，作为当前节点的答案~
这样搜一遍，正向的便被统计完了~

从LCA到终点的方法同理，只是在dep[i]-w[i]时可能有负数出现，加上30000的偏移量即可~

#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>

using namespace std;

const int N=300233;

struct player
{
    int s,t,lca,len;
}a[N];

int to[N<<1],nxt[N<<1],beg[N],tot;
int n,m,w[N],mdep;
vector<int> upper[N],down1[N],down2[N];

inline int read()
{
    int x=0;char ch=getchar();
    while(ch<'0' || '9'<ch)ch=getchar();
    while('0'<=ch && ch<='9')
    {
        x=(x<<1)+(x<<3)+(ch^48);
        ch=getchar();
    }
    return x;
}

inline void adde(int u,int v)
{
    to[++tot]=v;
    nxt[tot]=beg[u];
    beg[u]=tot;
}

inline void add(int u,int v)
{
    adde(u,v);adde(v,u);
}

int top[N],dep[N],fa[N],siz[N],son[N];
int pcnt[N],ans[N],delta[N<<1];

void dfs1(int u)
{
    if(dep[u]>mdep)mdep=dep[u];
    siz[u]=1;
    son[u]=0;

    for(int i=beg[u],v;i;i=nxt[i])
        if((v=to[i])!=fa[u])
        {
            dep[v]=dep[u]+1;
            fa[v]=u;

            dfs1(v);
            siz[u]+=siz[v];

            if(siz[v]>siz[son[u]] || !son[u])
                son[u]=v;
        }
}

void dfs2(int u)
{
    if(son[u])
    {
        top[son[u]]=top[u];
        dfs2(son[u]);

        for(int i=beg[u],v;i;i=nxt[i])
            if((v=to[i])!=fa[u] && v!=son[u])
            {
                top[v]=v;
                dfs2(v);
            }
    }
}

inline int lca(int a,int b)
{
    while(top[a]!=top[b])
    {
        if(dep[top[a]]<dep[top[b]])
            swap(a,b);
        a=fa[top[a]];
    }

    if(dep[a]>dep[b])
        return b;
    return a;
}

inline void dfs3(int u)
{
    int now=dep[u]+w[u],last=delta[now];

    for(int i=beg[u];i;i=nxt[i])
        if(to[i]!=fa[u])
            dfs3(to[i]);

    delta[dep[u]]+=pcnt[u];
    if(now<=mdep)
        ans[u]=delta[now]-last;

    for(int i=0;i<upper[u].size();i++)//reset belong roads
        delta[dep[upper[u][i]]]--;
}

inline void dfs4(int u)
{
    int now=dep[u]-w[u]+300000,last=delta[now];

    for(int i=beg[u];i;i=nxt[i])
        if(to[i]!=fa[u])
            dfs4(to[i]);

    for(int i=0;i<down1[u].size();i++)
        delta[down1[u][i]+300000]++;
    ans[u]+=delta[now]-last;
    for(int i=0;i<down2[u].size();i++)
        delta[down2[u][i]+300000]--;
}

int main()
{
    n=read();
    m=read();

    for(int i=1;i<n;i++)
        add(read(),read());

    dep[1]=1;
    fa[1]=1;
    dfs1(1);
    top[1]=1;
    dfs2(1);

    for(int i=1;i<=n;i++)
        w[i]=read();
    for(int i=1;i<=m;i++)
    {
        a[i].s=read(),a[i].t=read();
        a[i].lca=lca(a[i].s,a[i].t);
        a[i].len=dep[a[i].s]+dep[a[i].t]-(dep[a[i].lca]<<1);

        pcnt[a[i].s]++;
        upper[a[i].lca].push_back(a[i].s);
    }

    dfs3(1);

    for(int i=1;i<=m;i++)
    {
        down1[a[i].t].push_back(dep[a[i].t]-a[i].len);
        down2[a[i].lca].push_back(dep[a[i].t]-a[i].len);
    }

    memset(delta,0,sizeof(delta));
    dfs4(1);

    for(int i=1;i<=m;i++)
        if(dep[a[i].s]==dep[a[i].lca]+w[a[i].lca])
            ans[a[i].lca]--;

    printf("%d",ans[1]);
    for(int i=2;i<=n;i++)
        printf(" %d",ans[i]);

    return 0;
}