题目:http://codeforces.com/problemset
分析:倍增求lca,合并操作比较神奇,注意边界写法,有时包含顶端节点有时不包含,比如往上走2^0,包不包含他的父亲.
代码:
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int Tmax=100005,Tmax2=18;
struct node{
int c[15],num;
node(){num=0;}
void add(int x)
{
int p=lower_bound(c+1,c+1+num,x)-c,i;
if(p<=10){
for(i=10;i>=p;i--)
c[i]=c[i-1];
c[p]=x;
num++;
}
}
};
node g[Tmax2+5][Tmax];
int n,m,q,dep[Tmax],f[Tmax2+5][Tmax];
vector<int> G[Tmax];
node operator +(const node &a,const node &b)
{
node tmp;
int i=1,j=1;
tmp.num=0;
while((i<=a.num||j<=b.num)&&tmp.num<10)
{
tmp.num++;
if(i>a.num) tmp.c[tmp.num]=b.c[j],j++;
else if(j>b.num) tmp.c[tmp.num]=a.c[i],i++;
else if(a.c[i]<b.c[j]) tmp.c[tmp.num]=a.c[i],i++;
else tmp.c[tmp.num]=b.c[j],j++;
}
return tmp;
}
void dfs(int x,int d)
{
dep[x]=d;
int i,len=G[x].size();
for(i=0;i<len;i++)
if(dep[G[x][i]]==0)
{
f[0][G[x][i]]=x;
dfs(G[x][i],d+1);
}
return;
}
void init()
{
int i,j;
for(i=1;i<=Tmax2;i++)
for(j=1;j<=n;j++)
{
f[i][j]=f[i-1][f[i-1][j]];
g[i][j]=g[i-1][j]+g[i-1][f[i-1][j]];
}
return;
}
void work()
{
int i,j,u,v,a;
node ans;
for(i=1;i<=q;i++)
{
ans.num=0;
scanf("%d%d%d",&u,&v,&a);
if(dep[u]<dep[v]) swap(u,v);
for(j=Tmax2;j>=0;j--)
if(dep[f[j][u]]>=dep[v]){
ans=ans+g[j][u];
u=f[j][u];
}
for(j=Tmax2;j>=0;j--)
if(f[j][u]!=f[j][v])
{
ans=ans+g[j][u]+g[j][v];
u=f[j][u];
v=f[j][v];
}
ans=ans+g[0][u];
if(u!=v) ans=ans+g[0][f[0][u]]+g[0][v];
a=min(a,ans.num);
printf("%d ",a);
for(j=1;j<=a;j++)
printf("%d ",ans.c[j]);
printf("\n");
}
return;
}
int main()
{
int i,u,v;
scanf("%d%d%d",&n,&m,&q);
for(i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
for(i=1;i<=m;i++)
{
scanf("%d",&u);
g[0][u].add(i);
}
dfs(1,1);
init();
work();
return 0;
}