[树链剖分+线段树] bzoj4719: [Noip2016]天天爱跑步【留坑待填】

bzoj4719: [Noip2016]天天爱跑步：https://www.lydsy.com/JudgeOnline/problem.php?id=4719

从s到t
上升时，对于经过的每一个节点i
设经过了t[i]条路径 即用时t[i]
t[i]=dep[s]-dep[i] 得t[i]+dep[i]=dep[s]
同样 下降时，对于每一个经过的节点i
t[i]=(dep[i]-dep[lca])+(dep[s]-dep[lca])
得dep[i]-t[i]=2*dep[lca]-dep[s]
当i作为答案时 dep[i]-t[i]为定值
对于每一个权值建一颗动态开点的线段树
然后对于每一次查询就查询就查询这个权的线段树

然后这种O（nlog^2n） 而且内存巨大的破算法
居然在bzoj上跑过去了……跑过去了！
然后就在各种2s的OJ上T了……
现在不停课了感觉肝一道题都……好累啊
不搞了不搞了搞了我一个星期快死了
留坑待填

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
struct trnode
{
    int lc,rc,c,lazy;
}tr[24000000];int trlen;
struct node
{
    int x,y,next;
}a[610000];
int last[310000],len;
int fa[310000][25],dep[310000],son[310000],tot[310000],bin[25];
int ys[310000],id,top[310000];
int rt[1210000];
int n,m; 
void build(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;
}
void pre_node(int x)
{
    son[x]=0;tot[x]=1;
    for(int i=1;i<=20;i++)fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x][0])
        {
            fa[y][0]=x;
            dep[y]=dep[x]+1;
            pre_node(y);
            if(tot[son[x]]<tot[y]) son[x]=y;
            tot[x]+=tot[y];
        }
    }
}
int lca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=20;i>=0;i--) if(bin[i]<=dep[x]&&dep[fa[x][i]]>=dep[y]) x=fa[x][i];//不能跳过y
    if(x==y) return x;
    for(int i=20;i>=0;i--) if(bin[i]<=dep[x]&&fa[x][i]!=fa[y][i]) {x=fa[x][i];y=fa[y][i];}//跳到他们的最近公共祖先的儿子 
    return fa[x][0];
}
void pre_eage(int x,int tp)
{
    ys[x]=++id; top[x]=tp;
    if (son[x]!=0) pre_eage(son[x],tp);
    for (int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if (a[k].y!=fa[x][0]&&a[k].y!=son[x])
        {
            pre_eage(a[k].y,a[k].y);
        }
    }
}
void update(int x)
{
    int lc=tr[x].lc,rc=tr[x].rc;
    if (lc==0) lc=tr[x].lc=++trlen;
    tr[lc].lazy+=tr[x].lazy;tr[lc].c+=tr[x].lazy;
    if (rc==0) rc=tr[x].rc=++trlen;
    tr[rc].lazy+=tr[x].lazy;tr[rc].c+=tr[x].lazy;
    tr[x].lazy=0;
}
void change(int &x,int xl,int xr,int l,int r,int c)
{
    if (x==0) x=++trlen;//动态开点!
    if (xl==l&&xr==r)
    {
        tr[x].c+=c;tr[x].lazy+=c;
        return ;
    }
    int mid=(xl+xr)/2;
    if (tr[x].lazy) update(x); 
    if (r<=mid) change(tr[x].lc,xl,mid,l,r,c);
    else if (l>mid) change(tr[x].rc,mid+1,xr,l,r,c);
    else change(tr[x].lc,xl,mid,l,mid,c),change(tr[x].rc,mid+1,xr,mid+1,r,c);
}
int getsum(int x,int xl,int xr,int p)
{
    if (x==0) return 0;
    if (xl==xr) return tr[x].c;
    int mid=(xl+xr)/2;
    if (tr[x].lazy) update(x);
    if (p<=mid) return getsum(tr[x].lc,xl,mid,p);
    else return getsum(tr[x].rc,mid+1,xr,p); 
}
void solve(int x,int y,int u)
{
    int tx=top[x],ty=top[y];
    while(tx!=ty)
    {
        if(dep[tx]>dep[ty]) {swap(tx,ty);swap(x,y);}
        change(rt[u],1,n,ys[ty],ys[y],1);
        y=fa[ty][0]; ty=top[y];
    }
    if(x==y) change(rt[u],1,n,ys[x],ys[x],1);
    else
    {
        if(dep[x]>dep[y]) swap(x,y);
        change(rt[u],1,n,ys[x],ys[y],1);
    }
}
int w[310000];
int main()
{

    bin[0]=1;
    for (int i=1;i<=20;i++) bin[i]=(bin[i-1]<<1);
    scanf("%d%d",&n,&m);
    len=0;memset(last,0,sizeof(last));
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        build(x,y);
        build(y,x); 
    } 
    fa[1][0]=0;dep[1]=1;
    pre_node(1);
    id=0;pre_eage(1,1);
    for (int i=1;i<=n;i++) scanf("%d",&w[i]);
    for (int i=1;i<=m;i++)
    {
        int s,t;
        scanf("%d%d",&s,&t);
        int ll=lca(s,t);
        solve(s,ll,dep[s]);
        solve(t,ll,2*dep[ll]-dep[s]+600000);
        change(rt[dep[s]],1,n,ys[ll],ys[ll],-1);//lca多走了一次 
    }
    for (int i=1;i<=n;i++)
        printf("%d ",getsum(rt[dep[i]+w[i]],1,n,ys[i])+getsum(rt[dep[i]-w[i]+600000],1,n,ys[i]));//上升/下降时刚好能够到达这个点的定值  (有可能为负数)
    printf("\n");
    return 0; 
}