跟隔壁的雨天的尾巴异常相似
-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
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;
}
来源：https://www.cnblogs.com/lsyyy/p/11527322.html