基本定义
deep:深度
size:一个节点的子树的节点数
son:重儿子(size最大的)
fa:节点的父亲
top:节点所在链的最高节点
基本思路
第一次dfs找到节点所有的基本信息;
第二次dfs沿着重儿子造成重链,以其他节点造新的重链;
于是树就被转化为链了。
然后再用其他数据结构优化
为了使所有的链连在一起我们可以给所有的点一个编号。
原理
由于一个点在被查询的过程中可以走重链或轻链,而如果都走轻链也只会走O(logn)次。
例题
CF 326 E . Duff in the Army
在一棵n个节点的树中,有m个人(编号为1到m),给出每个人在树上的节点位置。有q个询问,对于每个询问v,u,a ,输出k,并从小到大输出该路径上编号最小的a个人的编号(若路径上不足a个人,有多少个人就输出多少个,k为输出的数量)。
1 ≤ n, m, q ≤ 10^5,1 ≤ a ≤ 10。
int dep[M],fa[M],sz[M],son[M],top[M];
int Re[M],id[M];
void dfs(int x,int pre=0){
sz[x]=1;dep[x]=dep[pre]+1;
for(int i=0;i<G[x].size();i++){
int y=G[x][i];
if(y!=pre){
fa[y]=x;
dfs(y,x);sz[x]+=sz[y];
if(sz[y]>sz[son[x]])son[x]=y;
}
}
}
void rdfs(int x,int f){
top[x]=f;
//给树上的点编号
Re[id[x]=++num]=x;
if(son[x])rdfs(son[x],f);
for(int i=0;i<G[x].size();i++){
int y=G[x][i];
if(y!=son[x]&&y!=fa[x])rdfs(y,y);
}
}
//如果要找树上的从a到b的信息需要一条一条链的找
void Query(int a,int b){
//不在一条链
while(top[a]!=top[b]){
//从最低的点开始跳
if(dep[top[a]]<dep[top[b]])swap(a,b);
//搜集整条链的信息
query(id[top[a]],id[a]);
//再跳出该链
a=fa[top[a]];
}if(id[a]>id[b])swap(a,b);
query(id[a],id[b]);
}
综上
#include<bits/stdc++.h>
#define PB(a,b) G[a].push_back(b)
using namespace std;
const int M=1e5+5;
vector<int>S[M],G[M];
int a,b,c,n,m,q,num;
int dep[M],fa[M],sz[M],son[M],top[M];
int Re[M],id[M],T[12][M<<2],mx,ans[M];
void dfs(int x,int pre=0){
sz[x]=1;dep[x]=dep[pre]+1;
for(int i=0;i<G[x].size();i++){
int y=G[x][i];
if(y!=pre){
fa[y]=x;
dfs(y,x);sz[x]+=sz[y];
if(sz[y]>sz[son[x]])son[x]=y;
}
}
}
void rdfs(int x,int f){
top[x]=f;
Re[id[x]=++num]=x;
if(son[x])rdfs(son[x],f);
for(int i=0;i<G[x].size();i++){
int y=G[x][i];
if(y!=son[x]&&y!=fa[x])rdfs(y,y);
}
}
void build(int L=1,int R=num,int p=1){
int &len=T[0][p];
if(L==R){
int nw=Re[L];
for(int i=0;i<S[nw].size();i++){
T[++len][p]=S[nw][i];
}
return;
}int mid=R+L>>1,p1=p<<1,p2=p<<1|1, a=1,b=1;
build(L,mid,p1);build(mid+1,R,p2);
int l1=T[0][p1],l2=T[0][p2];
while(len<10&&a<=l1&&b<=l2){
if(T[a][p1]<T[b][p2])T[++len][p]=T[a++][p1];
else T[++len][p]=T[b++][p2];
}
while(len<10&&a<=l1)T[++len][p]=T[a++][p1];
while(len<10&&b<=l2)T[++len][p]=T[b++][p2];
}
void query(int l,int r,int L=1,int R=num,int p=1){
if(r<L||R<l)return;
if(l<=L&&R<=r){
for(int i=1;i<=min(T[0][p],mx);i++)
ans[++ans[0]]=T[i][p];
return;
}int mid=L+R>>1;
query(l,r,L,mid,p<<1);
query(l,r,mid+1,R,p<<1|1);
}
inline void solve(){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
mx=c;ans[0]=0;
while(top[a]!=top[b]){
if(dep[top[a]]<dep[top[b]])swap(a,b);
query(id[top[a]],id[a]);
a=fa[top[a]];
}if(id[a]>id[b])swap(a,b);
query(id[a],id[b]);
sort(ans+1,ans+1+ans[0]);
ans[0]=min(ans[0],c);
for(int i=0;i<=ans[0];i++)printf("%d ",ans[i]);
puts("");
}
int main(){
scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<n;i++){
scanf("%d %d",&a,&b);
PB(a,b);PB(b,a);
}for(int i=1;i<=m;i++){
scanf("%d",&a);
if(S[a].size()<10)S[a].push_back(i);
}
dfs(1);
rdfs(1,1);
build();
while(q--)solve();
return 0;
}