LCA 【bzoj1787】[Ahoi2008]Meet 紧急集合

LCA 【bzoj1787】[Ahoi2008]Meet 紧急集合

题目链接:https://www.lydsy.com/JudgeOnline/problem.php?id=1787

注意到边权为一,x和y之间的距离为dep[x]+dep[y]-2*dep[LCA(x,y)]。

直接枚举三中LCA的情况即可。

我莫名其妙求了9次LCA,GXZ巨佬求了四次,emmm……

code

#include<iostream>
#include<cstdio>
using namespace std;
const int wx=1000017;
inline int read(){
int sum=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0';ch=getchar();}
return sum*f;
}
struct e{
int nxt,to;
}edge[wx*2];
int n,m,num;
int head[wx],f[wx][23],dep[wx];
void add(int from,int to){
edge[++num].nxt=head[from];
edge[num].to=to;
head[from]=num;
}
void dfs(int u,int fa){
dep[u]=dep[fa]+1;
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].to;
if(v==fa)continue;
f[v][0]=u;dfs(v,u);
}
}
void pre(){
for(int j=1;j<=21;j++){
for(int i=1;i<=n;i++){
f[i][j]=f[f[i][j-1]][j-1];
}
}
}
int LCA(int x,int y){
if(dep[x]<dep[y])swap(x,y);
for(int i=21;i>=0;i--){
if(dep[f[x][i]]>=dep[y])x=f[x][i];
}
if(x==y)return x;
for(int i=21;i>=0;i--){
if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
}
return f[x][0];
}
int find(int x,int y,int z){
int lca=LCA(x,y);
int lca2=LCA(lca,z);
return dep[x]+dep[y]-2*dep[lca]+dep[z]+dep[lca]-2*dep[lca2];
}
void work(int x,int y,int z){
int dis1=find(x,y,z);
int dis2=find(x,z,y);
int dis3=find(y,z,x);
int maxdis=min(dis1,min(dis2,dis3));
if(dis1==maxdis){
printf("%d %d\n",LCA(x,y),dis1);
}
else if(dis2==maxdis){
printf("%d %d\n",LCA(x,z),dis2);
}
else printf("%d %d\n",LCA(y,z),dis3);
}
int main(){
n=read();m=read();
for(int i=1;i<n;i++){
int x,y;x=read();y=read();
add(x,y);add(y,x);
}
dfs(1,0);pre();
for(int i=1;i<=m;i++){
int x,y,z;
x=read();y=read();z=read();
work(x,y,z);
}
return 0;
}
上一篇:【BZOJ1787】[Ahoi2008]Meet 紧急集合 LCA


下一篇:使用perf + FlameGraph生成进程火焰图