NOIP 2016 天天爱跑步

---

【题目分析】
分析一下每个人的路径，在树中可以分为上升的一段和下降的一段。那么对于上升的一段，它的深度+观察时间是一个定值的时候，才可能被观察到；对于下降的情况，它的观察时间减去深度是一个定值。所以我们统计每个点他可以观察到的值（分为上升和下降两部分进行讨论）。然后只需要树链剖分，打上nlogn个标记（可以用vector）。然后差分数组从前到后、跑一边就可以了。

---

【代码】（考场程序）

#include 
#include 
#include 
#include 
#include 
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]]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;iif (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]while (top[a]!=top[b])
    {
        if (dep[top[a]]pos[top[a]]].push_back(fn);
        ed[pos[a]+1].push_back(fn);
        a=fa[top[a]];
    }
    if (dep[a]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;js[st[i][j]]++;
        for (int j=0;js[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;js[st[i][j]]++;
        for (int j=0;js[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;iint 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;
}