天天愛跑步[lca][dfs序][線段樹動態開點]
阿新 • • 發佈:2018-11-04
大家都寫的差分,我太菜了看不懂啊
與是這成了我練習暴力資料結構的好機會...
我們發現,當往上走時,一個點對答案有貢獻, 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; }