思路：这个题其实就是树上的第k小，主席树的本质还是类似于前缀和一样的结构，所以是完全相同的，所以我们在树上也可以用同样的方法，我们对于每一个节点进行建树，然后和普通的树上相同，ab之间的距离是等于
root[a]+root[b]-root[lca[a,b]]-root[fa[lca[a,b]]]
代码：
复制代码 #include <bits/stdc++.h> using namespace std; const int maxn=1e5+7; const int POW=18; int num[maxn],sa[maxn]; int ls[maxn40],rs[maxn40]; int sum[maxn*40]; int root[maxn]; vector<int>mp[maxn]; int dis[maxn]; int p[maxn][POW]; int cnt; int f[maxn];
void build(int l,int r,int &rt) { rt=++cnt; sum[rt]=0; if(l>=r)return ; int mid=(l+r)>>1; build(l,mid,ls[rt]); build(mid+1,r,rs[rt]); } void update(int last,int p,int l,int r,int &rt) { rt=++cnt; ls[rt]=ls[last]; rs[rt]=rs[last]; sum[rt]=sum[last]+1; if(l>=r)return ; int mid=(l+r)>>1; if(p<=mid)update(ls[last],p,l,mid,ls[rt]); else update(rs[last],p,mid+1,r,rs[rt]); } int query(int lrt,int rrt,int lcart,int lcafrt,int l,int r,int k) { if(l>=r)return l; int mid=(l+r)>>1; int ans=sum[ls[rrt]]+sum[ls[lrt]]-sum[ls[lcart]]-sum[ls[lcafrt]]; if(k<=ans) return query(ls[lrt],ls[rrt],ls[lcart],ls[lcafrt],l,mid,k); else return query(rs[lrt],rs[rrt],rs[lcart],rs[lcafrt],mid+1,r,k-ans); } void dfs(int u,int fa,int tot) { f[u]=fa; dis[u]=dis[fa]+1; p[u][0]=fa; for(int i=1;i<POW;i++) p[u][i]=p[p[u][i-1]][i-1]; update(root[fa],num[u],1,tot,root[u]); for(int i=0;i<mp[u].size();i++){ int v=mp[u][i]; if(v==fa)continue; dfs(v,u,tot); } } int lca(int a,int b) { if(dis[a]>dis[b])swap(a,b); if(dis[a]<dis[b]){ int del=dis[b]-dis[a]; for(int i=0;i<POW;i++) if(del&(1<<i))b=p[b][i]; } if(a!=b){ for(int i=POW-1;i>=0;i--){ if(p[a][i]!=p[b][i]){ a=p[a][i];b=p[b][i]; } } a=p[a][0];b=p[b][0]; } return a; } int main() { int n,m; while(~scanf("%d%d",&n,&m)){ for(int i=0;i<n;i++)mp[i].clear(); memset(dis,0,sizeof(dis)); memset(p,0,sizeof(p)); memset(f,0,sizeof(f)); for(int i=1;i<=n;i++){ scanf("%d",&num[i]); sa[i]=num[i]; } cnt=0; sort(sa+1,sa+1+n); int tot=unique(sa+1,sa+1+n)-sa-1; for(int i=1;i<=n;i++){ num[i]=lower_bound(sa+1,sa+tot+1,num[i])-sa; } int a,b,c; for(int i=1;i<n;i++){ scanf("%d%d",&a,&b); mp[a].push_back(b); mp[b].push_back(a); } build(1,tot,root[0]); dfs(1,0,tot); for(int i=1;i<=m;i++){ scanf("%d%d%d",&a,&b,&c); int t=lca(a,b); int id=query(root[a],root[b],root[t],root[f[t]],1,tot,c); printf("%d\n",sa[id]); } } return 0; } /* 8 5 105 2 9 3 8 5 7 7 1 2 1 3 1 4 3 5 3 6 3 7 4 8 2 5 1 2 5 2 2 5 3 2 5 4 7 8 2 */
