codeforces587C Duff in the Army

LCA 问题，需要用到倍增（包括点上的信息），不然会T。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<queue>
#include<vector>
#include<iostream>
#include<string>
#include<set>
#include<map>
#include<algorithm>
#include<complex>
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000")
#define nn 100100
#define maxk 18
#define ll long long
#define ULL unsiged long long
#define mod 1000000007
#define inf oxfffffffffff
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1

inline int read()
{
    char ch;
    bool flag=false;
    int re=0;
    while(!(((ch=getchar())>='0'&&(ch<='9'))||(ch=='-')));
    if(ch!='-')
    {
        re*=10;
        re+=ch-'0';
    }
    else
        flag=true;
    while((ch=getchar())>='0'&&(ch<='9'))
    {
        re*=10;
        re+=ch-'0';
    }
    if(flag)
        re=-re;
    return re;
}

vector<int>e[nn];
vector<int>h[nn][maxk];
int p[nn][maxk],dep[nn];

void DFS(int u)
{
    for(auto &i:e[u])
    {
        int v=i;
        if(p[u][0]==i) continue;
        p[i][0]=u;
        dep[i]=dep[u]+1;
        DFS(i);
    }
}

void link(vector<int> &a,vector<int> &b,vector<int> &c,int lim)
{
    int i=0,j=0,temp;
    while(c.size()<lim && (i<a.size() || j<b.size()))
    {
        if(i<a.size() && (j>=b.size() || a[i]<b[j]))
        {
            c.push_back(a[i]);
            i++;
        }
        else if(j<b.size() && (i>=a.size() || a[i]>b[j]))
        {
            c.push_back(b[j]);
            j++;
        }
        else
        {
            c.push_back(a[i]);
            i++;j++;
        }
    }
}

void RMQ(int n)
{
    for(int j=1;j<maxk;j++)
    {
        for(int i=1;i<=n;i++)
        {
            p[i][j]=p[p[i][j-1]][j-1];
            link(h[i][j-1],h[p[i][j-1]][j-1],h[i][j],10);
        }
    }
}

int getlca(int u,int v)
{
    if(dep[u]<dep[v]) swap(u,v);
    for(int i=maxk-1;i>=0;i--)
        if(dep[u]-(1<<i)>=dep[v])
            u=p[u][i];
    if(u==v) return u;
    for(int i=maxk-1;i>=0;i--)
        if(p[u][i]!=p[v][i])
            u=p[u][i],v=p[v][i];
    return p[u][0];
}

void solve(int u,int hm,int k,vector<int> &ans)
{
    vector<int>temp;
    for(int i=maxk-1;i>=0;i--)
    {
        if((1<<i)<=hm)
        {
            temp.clear();
            link(ans,h[u][i],temp,k);
            ans=temp;
            hm-=(1<<i);
            u=p[u][i];
        }
    }
}

int main()
{
    int n,m,q;
    n=read();
    m=read();
    q=read();
    for(int i=1;i<n;i++)
    {
        int u=read();
        int v=read();

        e[u].push_back(v);
        e[v].push_back(u);
    }
    for(int i=1;i<=m;i++)
    {
        int u=read();
        if(h[u][0].size()<10) h[u][0].push_back(i);
    }
    DFS(1);
    RMQ(n);
    while(q--)
    {
        int a,b,c;
        a=read();
        b=read();
        c=read();
        int lca=getlca(a,b);
        vector<int>va,vb,vc;
        solve(a,dep[a]-dep[lca]+1,c,va);
        solve(b,dep[b]-dep[lca]+1,c,vb);
        link(va,vb,vc,c);
        printf("%d",vc.size());
        for(auto &i:vc)
            printf(" %d",i);
        printf("\n");
    }
    return 0;
}