Codeforces587 C.Duff in the Army（倍增LCA）

题意：

给定n个点的树，表示有n个城市。
有m个人，住在这n个城市中，第i个人住在城市ci中。
q次询问，每次询问给出u,v,a，问城市u到城市j路径上的所有人中前a小的编号。

数据范围：n,m,q<=1e5，a<=10

解法：

树上路径问题容易想到LCA,
观察到a<=10,那么在倍增LCA数组的每一格都存一个vector,
只存格子表示的范围中编号前10小的数即可.
预处理复杂度O(n*log*10)
单组询问O(log*10)
视为n,q同阶,总复杂度为O(n*log*10)

code：

#include <bits/stdc++.h>
using namespace std;
const int maxm=2e5+5;
const int maxd=20;
vector<int>g[maxm];
int n,m,q;
//
vector<int>idx[maxm][25];
int f[maxm][25];
int d[maxm];
int k=10;//前k小
//
vector<int> merged(vector<int>& a,vector<int>& b){
    int c=0,cc=0;
    int lena=a.size();
    int lenb=b.size();
    vector<int>ans;
    while((int)ans.size()<k&&c<lena&&cc<lenb){
        if(a[c]<=b[cc]){
            ans.push_back(a[c++]);
        }else{
            ans.push_back(b[cc++]);
        }
    }
    while((int)ans.size()<k&&c<lena){
        ans.push_back(a[c++]);
    }
    while((int)ans.size()<k&&cc<lenb){
        ans.push_back(b[cc++]);
    }
    return ans;
}
void dfs(int x,int fa){
    for(int j=1;j<=maxd;j++){
        f[x][j]=f[f[x][j-1]][j-1];
        idx[x][j]=merged(idx[x][j-1],idx[f[x][j-1]][j-1]);
    }
    for(int v:g[x]){
        if(v==fa)continue;
        d[v]=d[x]+1;
        f[v][0]=x;
        dfs(v,x);
    }
}
vector<int> solve(int x,int v){
    if(d[x]<d[v])swap(x,v);
    vector<int>ans;
    for(int i=maxd;i>=0;i--){
        if(f[x][i]&&d[f[x][i]]>=d[v]){
            ans=merged(ans,idx[x][i]);
            x=f[x][i];
        }
    }
    if(x==v){
        ans=merged(ans,idx[x][0]);
        return ans;
    }
    for(int i=maxd;i>=0;i--){
        if(f[x][i]!=f[v][i]){
            ans=merged(ans,idx[x][i]);
            ans=merged(ans,idx[v][i]);
            x=f[x][i];
            v=f[v][i];
        }
    }
    ans=merged(ans,idx[x][0]);
    ans=merged(ans,idx[v][0]);
    ans=merged(ans,idx[f[v][0]][0]);
    return ans;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    cin>>n>>m>>q;
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    for(int i=1;i<=m;i++){
        int x;cin>>x;
        idx[x][0].push_back(i);
    }
    dfs(1,1);
    while(q--){
        int x,v;cin>>x>>v>>k;
        vector<int>ans=solve(x,v);
        cout<<ans.size()<<' ';
        for(auto i:ans){
            cout<<i<<' ';
        }
        cout<<'\n';
    }
    return 0;
}

---