[BZOJ1787]AHOI2008 緊急集合|LCA

仔細想想就是水題,找出三個點兩兩之間的LCA,必然會出現至少有兩個LCA相同,因爲肯定要有兩個人先走到一起再往上走嘛,或者就是三個都相同了。。那麼走到不同的LCA就是要走到的點,因爲這個LCA只會使一個點走遠路嘛。。

#include<cstdio>
#define N 500005
#include<iostream>
using namespace std;
struct edge{
	int e,next;
} ed[N*2];
int n,m,s,e,c,i,ne=0,f1,f2,f3,a[N],fa[N][22],d[N];
void add(int s,int e)
{
	ed[++ne].e=e;ed[ne].next=a[s];a[s]=ne;
}
void dfs(int x)
{
	for (int j=1;j<21;j++) fa[x][j]=fa[fa[x][j-1]][j-1];
	for (int j=a[x];j;j=ed[j].next)
		if (ed[j].e!=fa[x][0])
		{
			fa[ed[j].e][0]=x;d[ed[j].e]=d[x]+1;
			dfs(ed[j].e);
		}
}
int lca(int x,int y)
{
	if (d[x]>d[y]) swap(x,y);
	for (int j=20;j>=0;j--) if (fa[y][j]&&d[fa[y][j]]>=d[x]) y=fa[y][j];
	for (int j=20;j>=0;j--) if (fa[x][j]!=fa[y][j]) x=fa[x][j],y=fa[y][j];
	if (x==y) return y; else return fa[y][0];
}
int main()
{
	scanf("%d%d",&n,&m);
	for (i=1;i<n;i++)
	{
		scanf("%d%d",&s,&e);
		add(s,e);add(e,s);
	}
	fa[1][0]=0;d[1]=0;dfs(1);
	for (i=1;i<=m;i++)
	{
		scanf("%d%d%d",&s,&e,&c);
		f1=lca(s,e);f2=lca(e,c);f3=lca(s,c);
		if (f1==f2)
		{
			printf("%d %d\n",f3,d[e]-d[f1]-d[f1]+d[s]-d[f3]+d[c]);
			continue;
		}
		else if (f1==f3)
		{
			printf("%d %d\n",f2,d[e]-d[f1]+d[s]-d[f1]+d[c]-d[f2]);
			continue;
		}
		else 
		{
			printf("%d %d\n",f1,d[s]-d[f2]+d[c]-d[f2]+d[e]-d[f1]);
			continue;
		}
	}
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章