图论题找结论
Analysis
思考,如果一条路径能够被观察员观察,最基本的条件就是观察员在这条路径上
那么对于一条路径,我们分别考虑观察员位于
x
−
>
l
c
a
x->lca
x−>lca和
y
−
>
l
c
a
y->lca
y−>lca路径上的情况
Case 1
如果观察员u位于
x
−
>
l
c
a
x->lca
x−>lca上,能够观察到这条路径,一定满足下列式子:
d
e
p
[
x
]
=
d
e
p
[
u
]
+
w
[
u
]
dep[x]=dep[u]+w[u]
dep[x]=dep[u]+w[u]
Case 2
如果观察员u位于
y
−
>
l
c
a
y->lca
y−>lca上,能够观察到这条路径,一定满足下列式子:
d
e
p
[
x
]
−
2
∗
d
e
p
[
l
c
a
]
=
w
[
u
]
−
d
e
p
[
u
]
dep[x]-2*dep[lca]=w[u]-dep[u]
dep[x]−2∗dep[lca]=w[u]−dep[u]
于是问题就变成了统计每个观察员的子树中分别满足上述式子的人数。
然后做法就很多了
I 线段树合并
时间复杂度O(n log n)
实现细节:
1.差分的使用,是在lca的父亲处-1
2.若观察员位于lca处,则会被统计重复(很多树上操作都需要注意这一点)
所以我们对于两个式子的处理稍有不同。以下给出的代码采用了在第一棵线段树上差分时,fa[lca]处-1,而第二棵则在lca处-1
#include<bits/stdc++.h>
#define in read()
#define re register
using namespace std;
inline int read(){
char ch;int f=1,res=0;
while((ch=getchar())<'0'||ch>'9') if(ch=='-') f=-1;
while(ch>='0'&&ch<='9'){
res=(res<<1)+(res<<3)+(ch^48);
ch=getchar();
}
return f==1?res:-res;
}
const int N=3e5+10;
int n,m,w[N];
int nxt[N<<1],head[N],to[N<<1],ecnt=0;
inline void add(int x,int y){
nxt[++ecnt]=head[x];head[x]=ecnt;to[ecnt]=y;
}
int dep[N],sze[N],son[N],top[N],fa[N];
void dfs1(int u,int fu){
fa[u]=fu;dep[u]=dep[fu]+1;sze[u]=1;
for(re int e=head[u],v;e;e=nxt[e])
if((v=to[e])!=fu) {
dfs1(v,u);
sze[u]+=sze[v];
if(sze[v]>sze[son[u]]) son[u]=v;
}
}
void dfs2(int u,int tp){
top[u]=tp;
if(son[u]) dfs2(son[u],tp);
for(re int e=head[u],v;e;e=nxt[e])
if((v=to[e])!=fa[u]&&v!=son[u])
dfs2(v,v);
}
inline int getlca(int x,int y){
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]) swap(x,y);
x=fa[top[x]];
}
return dep[x]<dep[y]?x:y;
}
struct seg{
int cnt[N*40],lc[N*40],rc[N*40],rt[N],tot;
void modify(int &k,int l,int r,int pos,int val){
if(!k) k=++tot;
if(l==r){
cnt[k]+=val;
return;
}
int mid=l+r>>1;
if(pos<=mid) modify(lc[k],l,mid,pos,val);
else modify(rc[k],mid+1,r,pos,val);
cnt[k]=cnt[lc[k]]+cnt[rc[k]];
}
int merge(int x,int y){
if(!x||!y) return x+y;
cnt[x]+=cnt[y];
lc[x]=merge(lc[x],lc[y]);
rc[x]=merge(rc[x],rc[y]);
return x;
}
int query(int k,int l,int r,int pos){
if(!k) return 0;
if(l==r) return cnt[k];
int mid=l+r>>1;
if(pos<=mid) return query(lc[k],l,mid,pos);
else return query(rc[k],mid+1,r,pos);
}
}t1,t2;
int R,ans[N],maxn;
void dfsans(int u){
for(re int e=head[u],v;e;e=nxt[e]){
if((v=to[e])==fa[u]) continue;
dfsans(v);
t1.rt[u]=t1.merge(t1.rt[u],t1.rt[v]);
t2.rt[u]=t2.merge(t2.rt[u],t2.rt[v]);
}
ans[u]=t1.query(t1.rt[u],0,R,dep[u]+w[u])+t2.query(t2.rt[u],0,R,w[u]-dep[u]+maxn);
}
int main(){
int size=100<<20;//40M
// //__asm__ ("movl %0, %%esp\n"::"r"((char*)malloc(size)+size));//调试用这个
__asm__ ("movq %0,%%rsp\n"::"r"((char*)malloc(size)+size));//提交用这个
//main函数代码
n=in;m=in;R=2*n;
for(re int i=1;i<n;++i){
int u=in,v=in;
add(u,v);add(v,u);
}
dfs1(1,0);dfs2(1,1);
for(re int i=1;i<=n;++i) w[i]=in,maxn=max(maxn,dep[i]);
for(re int i=1;i<=m;++i){
int x=in,y=in;
int lca=getlca(x,y);
t1.modify(t1.rt[x],0,R,dep[x],1);
t1.modify(t1.rt[fa[lca]],0,R,dep[x],-1);
t2.modify(t2.rt[y],0,R,dep[x]-2*dep[lca]+maxn,1);
t2.modify(t2.rt[lca],0,R,dep[x]-2*dep[lca]+maxn,-1);
}
dfsans(1);
for(re int i=1;i<=n;++i) printf("%d ",ans[i]);
exit(0);//必须用exit
}