P1600 天天爱跑步 线段树合并+lca

天天爱跑步

有点毒瘤的题目  我们观察一下性质  在路径 u,v 上 如果这个路径 对某个点j 有贡献j肯定得在路径 u,v上  也就是说 如果当前遍历的点在 lca(u,v)上方的话 那么 u,v路径是肯定没有贡献的   因此到达lca时我们要把贡献去掉

在说怎么算贡献    

设u为起始点  v为终止点  

如果u对于j有贡献   有     

如果v对于j有贡献  则设u,v路径的时间为 t 有   

又     

化简可得  

然后需要注意的几点：

1.当我们遍历的点为lca(u,v)的时候 显然上面两个情况都会有贡献  所以我们要在lca处先去掉一种情况  再在 fa[lca]处在去掉另一种 类似差分的思想

2.当w[j]=0时要特判一下  

#include<bits/stdc++.h>
using namespace std;
const int N = 3e5+100;
const int M = N*50;
int n,m;
int sum[M],ls[M],rs[M],rt[N],pot[M],tot,ct,w[N],dep[N],lg[N],fa[N][30];
int h[N],nex[N<<1],to[N<<1],cur,ans[N];
void add_edge(int x,int y){
	to[++cur]=y;nex[cur]=h[x];h[x]=cur;
}
inline int in(){
	int x=0;char c=0;
	while(c<'0'||c>'9') c=getchar();
	while(c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
	return x;
}
inline int newnode(){
	return ct?pot[ct--]:++tot;
}
inline void del(int x){
	sum[x]=ls[x]=rs[x]=0;
	pot[++ct]=x;
}
inline void pushup(int o){
	sum[o]=sum[ls[o]]+sum[rs[o]];	
}
void upd(int &o,int l,int r,int pos,int v){
	if(!o) o=newnode();
	if(l==r){
		sum[o]+=v;
		return;
	}
	int mid = l+r>>1;
	if(pos<=mid) upd(ls[o],l,mid,pos,v);
	else upd(rs[o],mid+1,r,pos,v);
	pushup(o);
}
int merge(int x,int y){
	if(!x||!y) return x+y;
	ls[x]=merge(ls[x],ls[y]);
	rs[x]=merge(rs[x],rs[y]); 
	sum[x]+=sum[y];
	return x;
}
int query(int o,int l,int r,int pos){
	if(!o) return 0;
	if(l==r) return sum[o];
	int mid = l+r>>1;
	if(pos<=mid) return query(ls[o],l,mid,pos);
	else return query(rs[o],mid+1,r,pos);
}
void dfs1(int u,int f){
	dep[u]=dep[fa[u][0]=f]+1;
	for(int i = 1; i <= lg[dep[u]]; i++) fa[u][i]=fa[fa[u][i-1]][i-1];
	for(int i = h[u]; i; i = nex[i]){
		int v = to[i];
		if(v==f) continue;
		dfs1(v,u);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	while(dep[x]>dep[y]) x=fa[x][lg[dep[x]-dep[y]]-1];
	if(x==y) return x;
	for(int i = lg[dep[x]-1]; i >= 0; i--) 
	if(fa[x][i]!=fa[y][i]) 
	x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
void dfs2(int u,int f){
	for(int i = h[u]; i; i = nex[i]){
		int v = to[i];
		if(v==f) continue;
		dfs2(v,u);
		rt[u]=merge(rt[u],rt[v]);	
	}
	if(w[u]&&n+dep[u]+w[u]<=2*n) 
	ans[u]+=query(rt[u],1,n<<1,n+dep[u]+w[u]);
	ans[u]+=query(rt[u],1,n<<1,n+dep[u]-w[u]);
}
int main(){
	n=in();m=in();
	for(int i = 1; i < n; i++){
		int x,y;
		x=in(),y=in();
		add_edge(x,y);add_edge(y,x);
	}
	for(int i = 1; i <= n; i++){
		w[i]=in();
		lg[i]=(1<<lg[i-1]==i)+lg[i-1];
	}
	dfs1(1,0);
	for(int i = 1; i <= m; i++){
		int x,y;
		x=in(),y=in();
		int lc=lca(x,y);
		upd(rt[x],1,n<<1,n+dep[x],1);
		upd(rt[y],1,n<<1,n+2*dep[lc]-dep[x],1);
		upd(rt[lc],1,n<<1,n+dep[x],-1);
		upd(rt[fa[lc][0]],1,n<<1,n+2*dep[lc]-dep[x],-1);
	}
	dfs2(1,0);
	for(int i = 1; i <= n; i++) printf("%d%c",ans[i],i==n?'\n':' '); 
	return 0;
}