bzoj3631 松鼠的新家 树链poi~分

填坑填坑……链接：http://www.lydsy.com/JudgeOnline/problem.php?id=3631

题意：每走一遍一个节点就要增加一次计数器，最后到达终点不需增加，输出按顺序游历整棵树每个节点计数器的值。

就是裸的树链剖分……唯一的坑点

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=300005;
struct node
{
    int from,to,next;
}edge[maxn<<1];
int head[maxn],tot;
void addedge(int u,int v)
{
    edge[++tot]=(node){u,v,head[u]};head[u]=tot;
}
int n,ord[maxn],top[maxn],dep[maxn],pos[maxn],id[maxn],cnt,son[maxn],siz[maxn],pa[maxn];
void dfs1(int root,int fa,int deep)
{
    son[root]=0;dep[root]=deep;siz[root]=1;
    for(int i=head[root];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(pa[root]!=v)
        {
            pa[v]=root;
            dfs1(v,root,deep+1);
            siz[root]+=siz[v];
            if(siz[v]>siz[son[root]])son[root]=v;
        }
    }
}
void dfs2(int root,int tp)
{
    top[root]=tp;pos[++cnt]=root;id[root]=cnt;
    if(son[root])dfs2(son[root],tp);
    for(int i=head[root];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v!=pa[root]&&v!=son[root])dfs2(v,v);
    }
}
#define mid ((l+r)>>1)
#define lc root<<1
#define rc root<<1|1
#define lson lc,l,mid
#define rson rc,mid+1,r
int sm[maxn<<2],st[maxn<<2];
void pushdown(int root,int l,int r)
{
    if(st[root])
    {
        st[lc]+=st[root];st[rc]+=st[root];
        sm[lc]+=st[root]*(mid-l+1);sm[rc]+=st[root]*(r-mid);
        st[root]=0;
    }
}
void update(int root)
{
    sm[root]=sm[lc]+sm[rc];
}
void modify(int root,int l,int r,int L,int R)
{
    if(L<=l&&r<=R)
    {
        sm[root]+=r-l+1;st[root]++;
        return;
    }
    pushdown(root,l,r);
    if(L<=mid)modify(lson,L,R);if(R>mid)modify(rson,L,R);
    update(root);
}
void modify(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);
        modify(1,1,n,id[fx],id[x]);
        x=pa[fx];fx=top[x];
    }
    if(dep[x]>dep[y])swap(x,y);
    modify(1,1,n,id[x],id[y]);
}
int query(int root,int l,int r,int pos)
{
    if(l==r)return sm[root];
    pushdown(root,l,r);
    if(pos<=mid)return query(lson,pos);
    else return query(rson,pos);
}
int haha()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&ord[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);
    }
    dfs1(1,-1,1);dfs2(1,1);
    for(int i=1;i<n;i++)modify(ord[i],ord[i+1]);
    for(int i=1;i<=n;i++)
    {
        int val=query(1,1,n,id[i]);
        if(i!=ord[1])val--;
        printf("%d
",val);
    }
}
int sb=haha();
int main(){;}

可能就是除了开头的那个点每个点最后结果都要$-1$s，因为中间那些点加了两次，最后那个点到达后不再计数。