bzoj3631: [JLOI2014]松鼠的新家 树链剖分||树形dp

题目大意:
每次对一条链进行区间修改,最后询问每个点的点权

树剖裸题,似乎dfs序也可以做?有时间再补估计不会写了
dp是类似差分的思路

//树剖
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=300000+10;
typedef long long ll;
int last[N],len=0,l[N],r[N],
ls[N<<2],rs[N<<2];ll tag[N<<2],sum[N<<2],a[N];
int dep[N],s[N],pos[N],bl[N],f[N],size=0,
n,q;
struct Edge{int to,next;Edge(int to=0,int next=0):to(to),next(next){}}e[N<<1];
void add_edge(int u,int v){e[++len]=Edge(v,last[u]);last[u]=len;}
void dfs1(int u) 
{
    s[u]=1;
    for(int i=last[u];i;i=e[i].next) {
        int id=e[i].to;
        if(id==f[u])continue;
        dep[id]=dep[u]+1;f[id]=u;
        dfs1(id);
        s[u]+=s[id];
    }
}
void dfs2(int u,int chain)
{
    bl[u]=chain;pos[u]=++size;l[u]=r[u]=size;
    int k=0;
    for(int i=last[u];i;i=e[i].next) {
        int id=e[i].to;
        if(dep[id]>dep[u]&&s[id]>s[k])k=id;
    }
    if(k==0)return;
    dfs2(k,chain);
    for(int i=last[u];i;i=e[i].next) {
        int id=e[i].to;
        if(dep[id]>dep[u]&&id!=k)dfs2(id,id);
    }r[u]=size;
}
void build(int k,int l,int r)
{
    ls[k]=l;rs[k]=r;sum[k]=tag[k]=0;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
}
void pushdown(int k)
{
    int len=(rs[k]-ls[k]+1);
    tag[k<<1]+=tag[k];
    tag[k<<1|1]+=tag[k];
    sum[k<<1]+=(len-(len>>1))*tag[k];
    sum[k<<1|1]+=(len>>1)*tag[k];
    tag[k]=0;
}
void change(int k,int ql,int qr,ll delta)
{
    if(ql==ls[k]&&qr==rs[k]){sum[k]+=delta*(rs[k]-ls[k]+1);tag[k]+=delta;return;}
    if(tag[k])pushdown(k);
    int mid=(ls[k]+rs[k])>>1;
    if(qr<=mid)change(k<<1,ql,qr,delta);
    else if(ql>mid)change(k<<1|1,ql,qr,delta);
    else {
        change(k<<1,ql,mid,delta);
        change(k<<1|1,mid+1,qr,delta);
    }
    sum[k]=sum[k<<1]+sum[k<<1|1];
}
ll querysum(int k,int ql,int qr)
{
    if(ls[k]==ql&&rs[k]==qr){return sum[k];}
    if(tag[k])pushdown(k);
    int mid=(ls[k]+rs[k])>>1;
    if(qr<=mid)return querysum(k<<1,ql,qr);
    else if(ql>mid)return querysum(k<<1|1,ql,qr);
    else return (querysum(k<<1,ql,mid)+querysum(k<<1|1,mid+1,qr));
}
void solveadd(int x,int y,ll val)
{
    while(bl[x]!=bl[y]) {
        if(dep[bl[x]]<dep[bl[y]])swap(x,y);
        change(1,pos[bl[x]],pos[x],val);
        x=f[bl[x]];
    }if(dep[x]>dep[y])swap(x,y);
    change(1,pos[x],pos[y],val);
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n)scanf("%lld",&a[i]);
    for(int x,y,i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add_edge(x,y);add_edge(y,x);
    }
    dfs1(1);dfs2(1,1);
    build(1,1,n);
    solveadd(a[1],a[1],1);
    for(int i=1;i<n;i++)
        {solveadd(a[i],a[i+1],1);solveadd(a[i],a[i],-1);}
    solveadd(a[n],a[n],-1);
    fo(i,1,n)printf("%lld\n",querysum(1,pos[i],pos[i]));
    return 0;
}