大家都写的差分,我太菜了看不懂啊
与是这成了我练习暴力数据结构的好机会...
我们发现,当往上走时,一个点对答案有贡献, dep[s] - dep[x] = time[x]
往下走 dep[t] - dep[x] = len - time[x]
于是我想,把dep在一层的放在一起,建一棵树
那怎么提取区间呢
其实区间就是 st[u] -- ed[u] (dfs序)
然后动态开点就可以了
#include<bits/stdc++.h>
#define N 600005
using namespace std;
int first[N],next[N*2],to[N*2],tot;
int dep[N],fa[N][20],st[N],ed[N],rt[N];
struct Node{int l,r,val;}t[N*40];
int n,m,tim[N],sign,num,ans[N];
struct Query{int s,t,lca;}q[N];
int read(){
int cnt=0;char ch=0;
while(!isdigit(ch))ch=getchar();
while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
return cnt;
}
void add(int x,int y){
next[++tot]=first[x],first[x]=tot,to[tot]=y;
}
void dfs(int u,int f){
for(int i=1;i<=18;i++)
fa[u][i]=fa[fa[u][i-1]][i-1];
st[u]=++sign;
for(int i=first[u];i;i=next[i]){
int t=to[i]; if(t==f) continue;
fa[t][0]=u , dep[t]=dep[u]+1;
dfs(t,u);
}ed[u]=sign;
}
int lca(int x,int y){
if(dep[x]<dep[y]) swap(x,y);
for(int i=18;i>=0;i--)
if(dep[fa[x][i]]>=dep[y])
x=fa[x][i];
if(x==y) return x;
for(int i=18;i>=0;i--)
if(fa[x][i]!=fa[y][i])
x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
void Update(int &o,int l,int r,int pos,int val){
if(!o) o=++num;
t[o].val+=val;
if(l==r) return;
int mid=l+r>>1;
if(pos<=mid) Update(t[o].l,l,mid,pos,val);
else Update(t[o].r,mid+1,r,pos,val);
}
int quary(int o,int l,int r,int L,int R){
if(L<=l && r<=R) return t[o].val;
int mid=l+r>>1 , ans=0;
if(L<=mid) ans+=quary(t[o].l,l,mid,L,R);
if(R>mid) ans+=quary(t[o].r,mid+1,r,L,R);
return ans;
}
void clear(){
for(int i=1;i<=num;i++)
t[i].l=t[i].r=t[i].val=0;
memset(rt,0,sizeof(rt)); num=0;
}
int main(){
n=read(),m=read();
for(int i=1;i<n;i++){
int x=read(),y=read();
add(x,y),add(y,x);
}
dep[1]=1,dfs(1,0);
for(int i=1;i<=n;i++) tim[i]=read();
for(int i=1;i<=m;i++){
q[i].s=read() , q[i].t=read();
q[i].lca = lca(q[i].s , q[i].t);
}
for(int i=1;i<=m;i++){
int pos=dep[q[i].s] , l=q[i].lca;
Update(rt[pos],1,n,st[q[i].s],1);
if(st[fa[l][0]]) Update(rt[pos],1,n,st[fa[l][0]],-1);
}
for(int i=1;i<=n;i++){
ans[i] += quary(rt[dep[i]+tim[i]],1,n,st[i],ed[i]);
}
clear();
for(int i=1;i<=m;i++){
int l=q[i].lca , len = dep[q[i].s] + dep[q[i].t] - dep[l]*2;
int pos = len - dep[q[i].t] + n;
Update(rt[pos],1,n,st[q[i].t],1);
Update(rt[pos],1,n,st[l],-1);
}
for(int i=1;i<=n;i++){
ans[i] += quary(rt[tim[i]-dep[i]+n],1,n,st[i],ed[i]);
}
for(int i=1;i<=n;i++) printf("%d ",ans[i]);
return 0;
}