比赛 20160229 评测结果 AAAAAAAAAA
题目名称 距离咨询 最终得分 100
用户昵称 mikumikumi 运行时间 0.441 s
代码语言 C++ 内存使用 6.88 MiB
提交时间 2016-02-29 21:20:33
显示代码纯文本
#include<cstdio>
#include<vector>
#include<cmath>
#include<iostream>
using namespace std;
const int SIZEN=40010;
vector<pair<int,int> > s[SIZEN];
int dep[SIZEN],dfn[SIZEN*2],dfslist=0,first[SIZEN];
int pos[SIZEN];
int ST[SIZEN*2][20];
int N,M;
void add(int fr,int to,int lo)
{
	s[fr].push_back(make_pair(to,lo));
	s[to].push_back(make_pair(fr,lo));
}
void dfs(int x,int fa)
{
	dfn[++dfslist]=x;
	first[x]=dfslist;
	for(int i=0;i<s[x].size();i++)
	{
		int v=s[x][i].first,w=s[x][i].second;
		if(v==fa) continue;
		dep[v]=dep[x]+w;
		pos[v]=pos[x]+1;
		dfs(v,x);
		dfn[++dfslist]=x;
	}
}
void make_ST()
{
	int t=int(log2(dfslist+0.0));
	for(int i=1;i<=dfslist;i++) ST[i][0]=dfn[i];
	for(int i=1;i<=t;i++)
	{
		for(int j=1;j<=dfslist;j++)
		{
			if(j+(1<<i)-1>dfslist) break;
			if(pos[ST[j][i-1]]<pos[ST[j+(1<<(i-1))][i-1]]) ST[j][i]=ST[j][i-1];
			else ST[j][i]=ST[j+(1<<(i-1))][i-1];
		}
	}
}
void prepare()
{
	dep[1]=0;pos[1]=0;
	dfs(1,0);
	make_ST();
}
int LCA(int x,int y)
{
	int l=first[x],r=first[y];
	if(l>r) swap(l,r);
	int m=int(log2(r-l+1.0));
	int a=ST[l][m],b=ST[r-(1<<m)+1][m];
	if(pos[a]<pos[b]) return a;
	return b;
}
int main()
{
	freopen("dquery.in","r",stdin);
	freopen("dquery.out","w",stdout);
	scanf("%d%d",&N,&M);
	char str[5];
	int fr,to,lo,K;
	for(int i=1;i<=M;i++)
	{
		scanf("%d%d%d%s",&fr,&to,&lo,str);
		add(fr,to,lo);
	}
	prepare();
	scanf("%d",&K);
	int x,y;
	for(int i=1;i<=K;i++)
	{
		scanf("%d%d",&x,&y);
		int a=LCA(x,y);
		int ans=dep[x]+dep[y]-2*dep[a];
		printf("%d\n",ans);
	}
	return 0;
}