P3258 [JLOI2014]松鼠的新家 (树链剖分+差分)

题目链接

分析：
题目每次都是让一条路径上的点修改，我们可以先用树链刨分，把树上的点转化成DFS序数组里面的区间， 题目就变成了DFS序数组的区间修改，且每次修改只是把某个区间的值都加1 ，这里用差分数组可以在O（1）时间完成修改，最后统计前缀和输出就好。

题目有个坑就是从第二次出发开始出发点是不需要加1的，且最后到达的点也不需要+1，我们可以先忽略这个 ，最后统计完前缀和再把除第一个点之外的答案-1.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 3e5+7;
const ll inf = 34359738370;
struct 
{
    int to,next;
}e[maxn<<1];
int head[maxn],num=1;
void add(int u,int v)
{
    e[num].to=v;
    e[num].next=head[u];
    head[u]=num++;
}
int size[maxn],son[maxn],in[maxn],pos[maxn],fa[maxn],cnt=0,dep[maxn];
int top[maxn];//链顶
void dfs1(int rt,int par)
{
    size[rt]=1;
    for(int i=head[rt];i;i=e[i].next)
    {
        int to=e[i].to;
        if(to == par) continue;
        fa[to]=rt;
        dep[to]=dep[rt]+1;
        dfs1(to,rt);
        size[rt]+=size[to];
        if(size[to] > size[son[rt]]) son[rt]=to;
    }
}
void dfs2(int rt,int t)
{
    in[rt]=++cnt;
    pos[cnt]=rt;
    top[rt]=t;
    if(son[rt]) dfs2(son[rt],t);
    for(int i=head[rt];i;i=e[i].next)
    {
        int to=e[i].to;
        if(to == fa[rt] || to==son[rt]) continue;
        dfs2(to,to);
    }
}
//用差分数组维护dfs序的区间修改
int a[maxn],d[maxn],n;
void updata_chain(int x,int y)
{
    int fx=top[x],fy=top[y];
    while(fx != fy)
    {
        if(dep[fx] < dep[fy]) swap(x,y),swap(fx,fy);
        d[in[fx]]++,d[in[x]+1]--;//[in[fx],in[x]] +1
        x=fa[fx];
        fx=top[x];
    }
    if(in[x]>in[y]) swap(x,y);
    d[in[x]]++,d[in[y]+1]--;//[in[x],in[y]] +1
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
    for(int i=1;i<n;i++)
    {
        int v,u;
        scanf("%d %d",&v,&u);
        add(v,u),add(u,v);
    }
    dfs1(1,1);
    dfs2(1,1);
    for(int i=2;i<=n;i++)
    {
        updata_chain(a[i],a[i-1]);
    }
    for(int i=1;i<=n;i++) d[i]+=d[i-1];
    for(int i=2;i<=n;i++) d[in[a[i]]]--;//中间的节点 对应的dfs序的位置多加了一次
    for(int i=1;i<=n;i++) printf("%d\n",d[in[i]]);
    return 0;
}