【NOIP 2016 提高组】天天爱跑步

Description

Solution1

设一个点的深度为$d_i$，巡查的时间为$W_i$，
一个人的路径起点为s，LCA为z，终点为t，总共用时T，
把路径拆成向上和向下两段，
那么，如果这个人上升到点x时他被看到，一定符合：$d_s=d_x+W_x$，
同样，当他下降时如被点x看到，一定符合：$d_t-T=d_x-W_x$
所以每个点的答案就是子树中的某个值出现的次数，（也可以用主席树）
当树为一条链的时候，这个直接用两个桶+链表即可，
于是对于其他情况，用链剖即可，

复杂度：$O(n\log(n))$

Solution2

我们发现，之所以在其他情况不能像链那样做，是因为这样做如果遇到分叉，就会把其他子树的信息带到当前还没有做的子树中，而对于一棵子树开一个桶又不行，
但是，我们发现，我们当前要求的，就是只有两个桶的那两个值，或者说就是这两个值的变化，
所以直接像一条链那样做，只是把做之前和做之后的值做一下差即可。

复杂度：$O(n)$

Code

//我打的是solution1

#include <iostream>
#include <cstdio>
#include <cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
#define efo(i,q) for(int i=A[q];i;i=B[i][0])
using namespace std;
const int N=400050;
int read(int &n)
{
    char ch=' ';int q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-')w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int m,n;
struct qqww
{
    int s,c,fa,lt,zx;
}a[N];
int B[2*N][2],A[N],B0;
int zx[N];
int S[N*2],T[2*N];
struct wqwq
{int s,v,nx;};
wqwq Bo[3][N*10];
int Ao[3][N],Bo0[3];
int W[N],ans[N];
bool z[N];
void link(int q,int w)
{
    B[++B0][0]=A[q];A[q]=B0;B[B0][1]=w;
    B[++B0][0]=A[w];A[w]=B0;B[B0][1]=q;
}
int dfs(int q,int fa,int c)
{
    a[q].fa=fa;a[q].c=c;a[q].s=1;
    efo(i,q)if(B[i][1]!=fa)a[q].s+=dfs(B[i][1],q,c+1);
    return a[q].s;
}
void dfs2(int q,int fa,int lt)
{
    a[q].lt=lt;zx[a[q].zx=++zx[0]]=q;
    int mx=0;
    efo(i,q)if(B[i][1]!=fa)if(a[B[i][1]].s>a[mx].s)mx=B[i][1];
    if(mx)dfs2(mx,q,lt);
    efo(i,q)if(B[i][1]!=fa&&B[i][1]!=mx)dfs2(B[i][1],q,B[i][1]);
}
void link(int k,int q,wqwq w)
{
    w.nx=Ao[k][q];
    Bo[k][++Bo0[k]]=w;
    Ao[k][q]=Bo0[k];
}
int putnotic(int q,int w,int q1,int w1)
{
    wqwq t;
    if(a[q].lt==a[w].lt)
    {
        if(a[q].c>a[w].c)
        {
            t.s=a[q1].c;t.v=1;
            link(1,q,t);
            t.v=-1;
            link(1,w,t);
            return w;
        }
        t.s=a[w1].c-(a[q1].c+a[w1].c-2*a[q].c);t.v=1;
        link(2,w,t);
        t.v=-1;
        link(2,q,t);
        return q;
    }
    if(a[a[q].lt].c>a[a[w].lt].c)
    {
        int lt=a[q].lt;
        t.s=a[q1].c;t.v=1;
        link(1,q,t);
        link(0,lt,t);
        return putnotic(a[lt].fa,w,q1,w1);
    }
    int lt=a[w].lt;
    int e=putnotic(q,a[lt].fa,q1,w1);
    t.s=a[w1].c-(a[q1].c+a[w1].c-2*a[e].c);t.v=1;
    link(2,w,t);
    link(0,lt,t);
    return e;
}
void dfs_ans(int q)
{
    for(q=zx[q];1;q=a[q].fa)
    {
        z[q]=1;
        for(int i=Ao[1][q];i;i=Bo[1][i].nx)S[Bo[1][i].s]+=Bo[1][i].v;
        for(int i=Ao[2][q];i;i=Bo[2][i].nx)T[N+Bo[2][i].s]+=Bo[2][i].v;
        ans[q]+=S[a[q].c+W[q]]+T[N+a[q].c-W[q]];
        if(q==a[q].lt)break;
    }
    for(int i=Ao[0][q];i;i=Bo[0][i].nx)T[N+Bo[0][i].s]=0,S[Bo[0][i].s]=0;
}
int main()
{
    freopen("running.in","r",stdin);
    freopen("running.out","w",stdout);
    int q,w,e;
    read(n),read(m);
    fo(i,1,n-1)
    {
        read(q),read(w);
        link(q,w);
    }
    dfs(1,0,1);dfs2(1,0,1);
    fo(i,1,n)read(W[i]);
    fo(i,1,m)
    {
        read(q),read(w);
        e=putnotic(q,w,q,w);
        if(a[q].c==W[e]+a[e].c)ans[e]++;
    }
    fod(i,n,1)if(!z[zx[i]])dfs_ans(i);
    fo(i,1,n)printf("%d ",ans[i]);
    return 0;
}