NOIP2016 T2 天天爱跑步

解题报告：

我们思考一下从x到y的路径，这个可以拆成从x到lca的路径和从lca到y的路径，这个很明显。

如果一个点i在从x到lca 的路径可以检测到的话，那么就有deep[i]+w[i]=deep[x]。

如果一个点i在从lca到y的路径上可以检测到的话，那么就有deep[i]-w[i]=deep[y]-t（t表示x到y的路径长度）。

对于每个点i，我们已知deep[i],w[i],只需要在路径上标上等号左侧的数字。

考虑树上差分，将问题转化为子树集问题。

dfs维护两个桶，一个向上的桶a和一个向下的桶b(分别表示deep[x],deep[y]-t)。

每次ans[x]的答案就是子树a[deep[x]+w[x]]+b[deep[x]-w[x]]的数量。

由于一个子树中节点的dfs序是一段连续的区间，答案用进入该节点时候的值减去出节点即可。

时间复杂度$O(n)$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
using namespace std;

int getint()
{
    int i=0;char c;
    for(c=getchar();c<'0'||c>'9';c=getchar());
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i;
}

const int N=3e5+5;
int n,m,tot;
int w[N],cnt[N<<2],ans[N];
int dep[N],fa[N][25];
int first[N],next[N<<1],to[N<<1];
struct node
{
    int dep,f;
    node(){}
    node(int _dep,int _f):
        dep(_dep),f(_f){}
};
vector<node>pos[N];

void add(int x,int y)
{
    next[++tot]=first[x],first[x]=tot,to[tot]=y;
}

inline void dfs(int u,int f)
{
    fa[u][0]=f;
    for(int i=1;i<=20;i++)fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int e=first[u];e;e=next[e])
    {
        int v=to[e];
        if(v!=f)
        {
            dep[v]=dep[u]+1;
            dfs(v,u);
        }
    }
}

int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=20;i>=0;i--)
        if(d&1<<i)x=fa[x][i];
    if(x==y)return x;
    for(int i=20;i>=0;i--)
        if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}

inline void solve(int u)
{
    int res=cnt[dep[u]+w[u]]+cnt[w[u]-dep[u]+3*n+1];
    for(int i=0;i<pos[u].size();i++)
        cnt[pos[u][i].dep]+=pos[u][i].f;
    for(int e=first[u];e;e=next[e])
    {
        int v=to[e];
        if(v!=fa[u][0])solve(v);
    }
    ans[u]=cnt[dep[u]+w[u]]+cnt[w[u]-dep[u]+3*n+1]-res;
}

int main()
{
    //freopen("lx.in","r",stdin);
    int x,y;
    n=getint(),m=getint();
    for(int i=1;i<n;i++)
    {
        x=getint(),y=getint();
        add(x,y),add(y,x);
    }
    dfs(1,0);
    for(int i=1;i<=n;i++)
        w[i]=getint();
    while(m--)
    {
        x=getint(),y=getint();
        int k=lca(x,y);

        pos[x].push_back(node(dep[x],1));
        pos[fa[k][0]].push_back(node(dep[x],-1));
        pos[y].push_back(node(dep[x]-2*dep[k]+3*n+1,1));
        pos[k].push_back(node(dep[x]-2*dep[k]+3*n+1,-1));   
    }
    solve(1);
    for(int i=1;i<=n;i++)
        cout<<ans[i]<<" ";
    return 0;
}