BZOJ3631 [JLOI2014]松鼠的新家（洛谷P3258）

树链剖分

BZOJ题目传送门
洛谷题目传送门

裸树剖不解释

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 300005
#define F inline
using namespace std;
struct tree{ int l,r,x,f; }t[N<<2];
struct edge{ int nxt,to; }ed[N<<1]; 
int n,k,ti,h[N],a[N],tp[N],sz[N],to[N],fa[N],id[N],in[N],dep[N],ans[N];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
F void writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
F void _write(int x){ writec(x),puts(""); }
#define addedge(x,y) ed[++k]=(edge){h[x],y},h[x]=k
void dfs1(int x){
    sz[x]=1,dep[x]=dep[fa[x]]+1;
    for (int i=h[x],v;i;i=ed[i].nxt)
        if ((v=ed[i].to)!=fa[x]){
            fa[v]=x,dfs1(v),sz[x]+=sz[v];
            if (sz[v]>sz[to[x]]) to[x]=v;
        }
}
void dfs2(int x){
    if (to[in[id[x]=++ti]=x]) tp[to[x]]=tp[x],dfs2(to[x]);
    for (int i=h[x],v;i;i=ed[i].nxt)
        if ((v=ed[i].to)!=fa[x]&&v!=to[x])
            tp[v]=v,dfs2(v);
}
F void pshd(int x){
    t[x<<1].f+=t[x].f,t[x<<1|1].f+=t[x].f;
    t[x<<1].x+=(t[x<<1].r-t[x<<1].l+1)*t[x].f;
    t[x<<1|1].x+=(t[x<<1|1].r-t[x<<1|1].l+1)*t[x].f;
    t[x].f=0;
}
void build(int x,int l,int r){
    t[x].l=l,t[x].r=r;
    if (l==r) return; int mid=l+r>>1;
    build(x<<1,l,mid),build(x<<1|1,mid+1,r);
}
void mdfy(int x,int l,int r,int w){
    if (t[x].l>r||t[x].r<l) return;
    if (t[x].l>=l&&t[x].r<=r){
        t[x].x+=(t[x].r-t[x].l+1)*w;
        t[x].f+=w; return;
    }
    if (t[x].f) pshd(x);
    mdfy(x<<1,l,r,w),mdfy(x<<1|1,l,r,w);
    t[x].x=t[x<<1].x+t[x<<1|1].x;
}
void srch(int x){
    if (t[x].l==t[x].r){ ans[in[t[x].l]]=t[x].x; return; }
    if (t[x].f) pshd(x); srch(x<<1),srch(x<<1|1);
}
F void nsrt(int x,int y){
    while (tp[x]!=tp[y]){
        if (dep[tp[x]]<dep[tp[y]]) swap(x,y);
        mdfy(1,id[tp[x]],id[x],1),x=fa[tp[x]];
    }
    if (dep[x]<dep[y]) swap(x,y);
    mdfy(1,id[y],id[x],1);
}
int main(){
    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(),addedge(x,y),addedge(y,x);
    dfs1(1),tp[1]=1,dfs2(1),build(1,1,n);
    for (int i=1;i<n;i++)
        nsrt(a[i],a[i+1]);
    srch(1);
    for (int i=2;i<=n;i++) ans[a[i]]--;
    for (int i=1;i<=n;i++) _write(ans[i]);
    return 0;
}