【BZOJ】3631 [JLOI2014]松鼠的新家 LCA+树上差分

题目传送门

再水一篇blog，可能也是我专门去学树上差分的最后一题吧。

一开始拿到题目，并没有想到树上差分的做法，看了题解之后：原来差分还可以这么用的啊，又学到了一种黑科技。

对于相邻的两个房间，直接和“【BZOJ】4390 [Usaco2015 dec]Max Flow”这题一样差分就好了。（真的就好了吗？）

重新考虑一下，发现从第$i$个房间走到第$i+1$个房间和从第$i+1$个房间走到第$i+2$个房间是同一条路径，但是第$i+1$个房间被计算了两次。

那么在此考虑，是不是除了起点外的房间都多计算了一次呢？的确如此。

于是直接把除了起点外的所有点都打上$-1$的标记再$O(n)$遍历一遍就行了。

附上AC代码：

#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;

const int N=3e5+10;
struct side{
    int to,nt;
}s[N<<1];
int n,x,y,num,h[N],t,sum[N],a[N];
int d[N],sz[N],hs[N],f[N],top[N];

inline char nc(){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
    return (void)(a*=f);
}

inline void add(int x,int y){
    s[++num]=(side){y,h[x]},h[x]=num;
    s[++num]=(side){x,h[y]},h[y]=num;
}

inline void so1(int x,int fa){
    d[x]=d[f[x]=fa]+1,sz[x]=1;
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=fa){
            so1(s[i].to,x),sz[x]+=sz[s[i].to];
            if (sz[s[i].to]>sz[hs[x]]) hs[x]=s[i].to;
        }
    return;
}

inline void so2(int x,int fa){
    top[x]=fa;
    if (hs[x]) so2(hs[x],fa);
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=f[x]&&s[i].to!=hs[x]) so2(s[i].to,s[i].to);
    return;
}

inline int query(int x,int y){
    for (int fx=top[x],fy=top[y]; fx!=fy; x=f[fx],fx=top[x])
        if (d[fx]<d[fy]) swap(fx,fy),swap(x,y);
    return d[x]<d[y]?x:y;
}

inline void so(int x){
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=f[x]) so(s[i].to),sum[x]+=sum[s[i].to];
    return;
}

int main(void){
    read(n);
    for (int i=1; i<=n; ++i) read(a[i]);
    for (int i=1; i<n; ++i) read(x),read(y),add(x,y);
    so1(1,0),so2(1,1);
    for (int i=1; i<n; ++i){
        t=query(a[i],a[i+1]);
        ++sum[a[i]],++sum[a[i+1]],--sum[t],--sum[f[t]];
    }
    so(1);
    for (int i=2; i<=n; ++i) --sum[a[i]];
    for (int i=1; i<=n; ++i) printf("%d ",sum[i]);
    return 0;
}