Codeforces 587C【树上倍增】

题意：
求树上两点路径中的前 a(<=10) 小的点权值。
思路：
类似LCA倍增算法来存储 i -> 2^j 路上的 前(<=10)小个 的点权值。
然后要写一个权值合并>>>
具体求 u->v 的话，就是先分别计算 {u->LCA} 和 {v->(LCA-1)} (减1是LCA会重叠), 然后再合并，输出答案。

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long LL;

#define mem(a,b) memset(a, b, sizeof(a))
#define LMid Left, Mid
#define RMid Mid+1,Right
#define Lson num<<1, Left, Mid
#define Rson num<<1|1, Mid + 1, Right


const int INF = 0x3f3f3f3f;
struct asd{
    int a[11];
    asd(){memset(a,INF,sizeof(a));}
    void Ins(int val){a[10] = val;sort(a, a+11);}
}w[17][100010];
struct Edge{int v, nex;}edge[200010];
int head[100010],tol;
int n, m, q, p[17][100010], dep[100010];
void init(){tol=0;mem(head,-1);}
void add(int u,int v){
    edge[tol] = (Edge){v, head[u]}, head[u] = tol++;
    edge[tol] = (Edge){u, head[v]}, head[v] = tol++;
}
asd Merge(asd x, asd y){
    asd ans = x;
    for(int i=0;i<10;i++)
        ans.Ins(y.a[i]);
    return ans;
}

void DFS(int u, int fa){
    int v;

    p[0][u] = fa;
    for(int i=1;i<17;i++){
        p[i][u] = p[i-1][p[i-1][u]];
        w[i][u] = Merge(w[i-1][p[i-1][u]], w[i-1][u]);
    }

    for(int i=head[u];~i;i=edge[i].nex){
        v = edge[i].v;
        if(v == fa) continue;
        dep[v] = dep[u] + 1;
        DFS(v, u);
    }
}

int LCA(int u, int v){
    if(dep[v] > dep[u]) swap(u, v);
    int h = dep[u] - dep[v];
    for(int i=0;(1<<i)<=h;i++)
        if(h & (1<<i)) u = p[i][u];
    if(u == v) return u;
    for(int i=16;i>=0;i--)
        if(p[i][u] != p[i][v]) v = p[i][v], u = p[i][u];
    return p[0][u];
}

asd getans(int u, int dp){
    asd ans;
    for(int i=16;i>=0;i--)
        if(dp & (1<<i)) ans = Merge(ans, w[i][u]), u = p[i][u];
    return ans;
}

int main(){
    int u, v, k, lca, tmp;
    asd ans;
    scanf("%d%d%d",&n,&m,&q);
    init();
    for(int i=2;i<=n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    for(int i=1;i<=m;i++){
        scanf("%d", &v);
        w[0][v].Ins(i);
    }
    DFS(1, 0);
    while(q--){
        ans;
        scanf("%d%d%d",&u,&v,&k);
        lca = LCA(u, v);
        ans = getans(u, dep[u] - dep[lca]);
        ans = Merge(ans, getans(v, dep[v] - dep[lca] + 1));
        tmp = 0;

        while(tmp < k && ans.a[tmp] <= m) tmp++;
        printf("%d",tmp);
        for(int i=0;i<tmp;i++)
            printf(" %d", ans.a[i]);
        puts("");
    }
    return 0;
}