天天爱跑步noip2016

洛谷

跟隔壁的雨天的尾巴异常相似

-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

1.Analysis

某条路线对当前节点x有贡献时只可能是

x在s到t的路线上（即s到lca到t）

那么首先我们可以对从s到t的路线划分为

从s到lca(s,t)

以及从t到lca

a.from s to lca

要有贡献

易得dep[s]-dep[x]=val[x](出现时间)

观察可得寻找s满足dep[s]=dep[x]+val[x]的数量

b.from lca to t

要有贡献

易得dep[s]+dep[x]-dep[lca]-dep[lca]=val[x](出现时间)

观察可得寻找s满足dep[s]-dep[lca]-dep[lca]=val[x]-dep[x]的数量

为了计算.我们规定lca被算在a部分中

2.实现

a.不断从叶子到lca累加

易联想到差分

b.由于种类不同可以开权值线段树

在递归回fa时合并线段树

3.问题

炸空间

4.解决

就权值线段树合并后空余节点进行一波垃圾回收

#include<bits/stdc++.h>
#define re return
#define inc(i,l,r) for(int i=l;i<=r;++i)
using namespace std;
template<typename T>inline void rd(T&x)
{
    char c;bool f=0;
    while((c=getchar())<'0'||c>'9')if(c=='-')f=1;
    x=c^48;
    while((c=getchar())>='0'&&c<='9')x=x*10+(c^48);
    if(f)x=-x;
}

const int maxn=300005,maxz=600000,maxl=300000;
int n,m,k,hd[maxn],ans[maxn],val[maxn];
struct node
{
    int to,nt;
}e[maxn<<1];

inline void add(int x,int y)
{
    e[++k].to=y;e[k].nt=hd[x];hd[x]=k;
    e[++k].to=x;e[k].nt=hd[y];hd[y]=k;
}

struct solu{
    //分为s,t两部分
    int top,tot,cnt;
    int rt[maxn*30],sum[maxn*30][2],ls[maxn*30],rs[maxn*30],first[maxn];
    int rab[maxn*30];
    struct ll{
        int flag,op,val,nt;
    }st[maxn<<2];
    
    inline void insert(int x,int y,int z,int f)
    {
        st[++top]=(ll){f,z,y,first[x]};
        first[x]=top;
    }
    
    
    inline int New()
    {
        int now;
        if(tot)now=rab[tot--];
        else now=++cnt;
        ls[now]=rs[now]=sum[now][0]=sum[now][1]=0;
        re now;
    }
    inline void Throw(int x)
    {
        rab[++tot]=x;
    }
    
     
    inline int query(int rt,int l,int r,int pos,int f)
    {
        if(!rt)re 0;
        if(l==r)
            re sum[rt][f];
        int mid=(l+r)>>1;
        if(pos<=mid)re query(ls[rt],l,mid,pos,f);
        else re query(rs[rt],mid+1,r,pos,f);
    }
    
    inline void add(int &rt,int l,int r,int pos,int vvl,int f)
    {
        if(!rt) rt=New();
        if(l==r)
        {
            if(l==299999)
            f=f;
            
            sum[rt][f]+=vvl;
            re ;
        }
        int mid=(l+r)>>1;
        if(pos<=mid)add(ls[rt],l,mid,pos,vvl,f);
        else add(rs[rt],mid+1,r,pos,vvl,f);
    }
    
    inline int merge(int x,int y,int l,int r)
    {
        if(!x||!y)re x+y;
        if(l==r)
        {
            if(l==299999)
            x=x;
            sum[x][0]+=sum[y][0];
            sum[x][1]+=sum[y][1];
            Throw(y);
            re x;
        }
        
        int mid=(l+r)>>1;
        
        ls[x]=merge(ls[x],ls[y],l,mid);
        rs[x]=merge(rs[x],rs[y],mid+1,r);
        Throw(y);
        re x;
    }
}T;

struct Tree_lca
{
    int top[maxn],size[maxn],son[maxn],dep[maxn],fa[maxn];
    
    inline void dfs(int x)
    {
        dep[x]=dep[fa[x]]+(size[x]=1);
        for(int i=hd[x];i;i=e[i].nt)
        {
            int v=e[i].to;
            if(v!=fa[x])
            {
                fa[v]=x;
                dfs(v);
                size[x]+=size[v];
                if(size[v]>size[son[x]])son[x]=v;
            }
        }
    }
    
    inline void dfs2(int x,int topf)
    {
        top[x]=topf;
        if(son[x])
        {
            dfs2(son[x],topf);
            for(int i=hd[x];i;i=e[i].nt)
            {
                int v=e[i].to;
                if(!top[v])
                dfs2(v,v);
            }
        }
    }
    
    inline int Lca(int x,int y)
    {
        while(top[x]!=top[y])
        {
            if(dep[top[x]]<dep[top[y]])x^=y^=x^=y;
            x=fa[top[x]];
        }
        re dep[x]<dep[y]?x:y;
    }
        
}S;

inline void dfs(int x)
{
    for(int i=hd[x];i;i=e[i].nt)
    {
        int v=e[i].to;
        if(v!=S.fa[x])
        {
            dfs(v);
            if(x==1&&v==2)
            x=1; 
            T.rt[x]=T.merge(T.rt[x],T.rt[v],1,maxz);
        }
    }
    if(x==1)
    x=1;
/**/    for(int i=T.first[x];i;i=T.st[i].nt)
        T.add(T.rt[x],1,maxz,T.st[i].val,T.st[i].op,T.st[i].flag);
    
    ans[x]=T.query(T.rt[x],1,maxz,S.dep[x]+val[x]+maxl,0);
    ans[x]+=T.query(T.rt[x],1,maxz,val[x]-S.dep[x]+maxl,1);
    
}

int main()
{
    freopen("in.txt","r",stdin);
    int x,y,z;
    rd(n),rd(m);
    inc(i,2,n)
    {
        rd(x),rd(y);
        add(x,y); 
    }
    
    S.dfs(1);
    S.dfs2(1,n+1); 
    
    inc(i,1,n)
    rd(val[i]);
    
    inc(i,1,m)
    {
        rd(x),rd(y);
        int lca=S.Lca(x,y),flca=S.fa[lca];
        T.insert(x,S.dep[x]+maxl,1,0);T.insert(flca,S.dep[x]+maxl,-1,0);
        T.insert(y,S.dep[x]-(S.dep[lca]<<1)+maxl,1,1);T.insert(lca,S.dep[x]-(S.dep[lca]<<1)+maxl,-1,1);        
    }
    dfs(1);
    
    inc(i,1,n)
    printf("%d ",ans[i]);
    re 0;
}