BZOJ 3631: [JLOI2014]松鼠的新家【差分】【LCA】

Description

松鼠的新家是一棵树，前几天刚刚装修了新家，新家有n个房间，并且有n-1根树枝连接，每个房间都可以相互到达，且俩个房间之间的路线都是唯一的。天哪，他居然真的住在“树”上。松鼠想邀请小熊维尼前来参观，并且还指定一份参观指南，他希望维尼能够按照他的指南顺序，先去a1，再去a2，……，最后到an，去参观新家。

可是这样会导致维尼重复走很多房间，懒惰的维尼不听地推辞。可是松鼠告诉他，每走到一个房间，他就可以从房间拿一块糖果吃。维尼是个馋家伙，立马就答应了。

现在松鼠希望知道为了保证维尼有糖果吃，他需要在每一个房间各放至少多少个糖果。因为松鼠参观指南上的最后一个房间an是餐厅，餐厅里他准备了丰盛的大餐，所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。

题解

这是第一次做对于点的差分。Lynstery大佬说LCA需要另外统计。由于每个点会被多统计一次，所以最后输出的时候除了起点都要-1。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 300006
using namespace std;
inline char nc(){
    static char buf[100000],*i=buf,*j=buf;
    return i==j&&(j=(i=buf)+fread(buf,1,100000,stdin),i==j)?EOF:*i++;
}
inline int _read(){
    char ch=nc();int sum=0;
    while(!(ch>='0'&&ch<='9'))ch=nc();
    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
    return sum;
}
int n,tot,sum[maxn],dep[maxn],tag[maxn],a[maxn],lnk[maxn],son[maxn*2],nxt[maxn*2],f[maxn][20];
bool vis[maxn];
void add(int x,int y){
    nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;
}
void dfs(int x){
    vis[x]=0;
    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]){
        dep[son[j]]=dep[x]+1;f[son[j]][0]=x;
        dfs(son[j]);
    }
}
void make_f(){
    for(int j=1;j<=19;j++)
     for(int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1];
}
int get(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    for(int j=19;j>=0;j--)if(dep[x]-(1<<j)>=dep[y])x=f[x][j];
    for(int j=19;j>=0;j--)if(f[x][j]!=f[y][j])x=f[x][j],y=f[y][j];
    if(x!=y)return f[x][0];
    return x;
}
void dfs2(int x){
    vis[x]=0;
    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]){
        dfs2(son[j]);
        tag[x]+=tag[son[j]];
    }
}
int main(){
    freopen("candy.in","r",stdin);
    freopen("candy.out","w",stdout);
    n=_read();
    for(int i=1;i<=n;i++)a[i]=_read();
    for(int i=1,x,y;i<n;i++)x=_read(),y=_read(),add(x,y),add(y,x);
    memset(vis,1,sizeof(vis));
    dfs(1);make_f();
    for(int i=2;i<=n;i++){
        int x=get(a[i-1],a[i]);
        tag[a[i-1]]++;tag[a[i]]++;tag[x]-=2;sum[x]++;
    }
    memset(vis,1,sizeof(vis));
    dfs2(1);
    for(int i=1;i<=n;i++) if(i==a[1])printf("%d\n",tag[i]+sum[i]);else
                                     printf("%d\n",tag[i]-1+sum[i]);
}