比赛 ZLXOI2015Day2 评测结果 RRRRRRRRRR
题目名称 ZLX的陨落 最终得分 0
用户昵称 wmez 运行时间 1.002 s
代码语言 C++ 内存使用 25.48 MiB
提交时间 2015-10-30 20:35:14
显示代码纯文本
#include<cstdio>
#include<vector>
using namespace std;
#define mp make_pair
vector <pair<int,int> >edge[100005];
int fa[100005][31];
int dis[100005][31];
int deep[100005];
int n,q,x,y,z,m;
void DFS(int x)
{
	for(int i=0;i<edge[x].size();i++)
	{
		if(deep[edge[x][i].first])
		{
			continue;
		}
		deep[edge[x][i].first]=deep[x]+1;
		fa[edge[x][i].first][0]=x;
		dis[edge[x][i].first][0]=edge[x][i].second;
		DFS(edge[x][i].first);
	}
}
int LCA(int x,int y)
{
	if(deep[x]<deep[y])
	{
		x^=y^=x^=y;
	}
	int i;
	for(i=0;(1<<i)<=deep[x];i++);
	i--;
	int res=0;
	for(int j=i;~j;j--)
	{
		if(deep[x]-(1<<j)>=deep[y])
		{
			res+=dis[x][j];
			x=fa[x][j];
		}
	}
	if(x==y)
	{
		return res;
	}
	for(int j=i;~j;j--)
	{
		if(fa[x][j]!=fa[y][j])
		{
			res+=dis[x][j]+dis[y][j];
			x=fa[x][j];
			y=fa[y][j];
		}
	}
	res+=dis[x][0]+dis[y][0];
	return res;
}
int main()
{
	freopen("ThefallingofZLX.in","r",stdin);
	freopen("ThefallingofZLX.ans","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		edge[x].push_back(mp(y,z));
		edge[y].push_back(mp(x,z));
	}
	deep[1]=1;
	DFS(1);
	for(int j=1;(1<<j)<=n;j++)
	{
		for(int i=1;i<=n;i++)
		{
			dis[i][j]=dis[i][j-1]+dis[fa[i][j-1]][j-1];
			fa[i][j]=fa[fa[i][j-1]][j-1];
		}
	}
	scanf("%d",&q);
	while(q--)
	{
		scanf("%d%d",&x,&y);
		printf("%d\n",LCA(x,y));
	}
	fclose(stdin);
	fclose(stdout);
	return 0;
}