POJ1986,作为LCA模板

// Note:Your choice is C++ IDE
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int mmax=100050;
typedef struct sdlfj
{
	int u,v,next,w;
}Edge;
typedef struct gf
{
	int u,v,num; 
}Query;
typedef struct sldf
{
	int v,num;
}Node;
vector<Node>node[mmax];
Edge edge[mmax];
Query query[mmax];
int set[mmax],d[mmax],head[mmax],check[mmax],ans[mmax];
int Enum,n,m,q;
void add(int u,int v,int w)
{
	edge[Enum].u=u;
	edge[Enum].v=v;
	edge[Enum].w=w;
	edge[Enum].next=head[u];
	head[u]=Enum++;
}
void insert(int u,int v,int w)
{
	add(u,v,w);
	add(v,u,w);
}
int findset(int x)
{
	if(set[x]==x)return x;
	return set[x]=findset(set[x]);
}
void Union(int x,int y)
{
	int a=findset(x);
	int b=findset(y);
	if(a!=b)
	set[a]=b;
}
void LCA_Tarjan(int u,int fa)
{
	int i;
	set[u]=u;
	for(i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;
		if(v==fa)continue;
		d[v]=d[u]+edge[i].w;
		LCA_Tarjan(v,u);
		Union(v,u);
	}
	check[u]=1;
	for(i=0;i<node[u].size();i++)//LCA(U,V);
	{
		ans[node[u][i].num]=findset(node[u][i].v);
	} 
}
void function()
{
	int i;
	memset(check,0,sizeof(check));
	for(i=1;i<=n;i++)
		if(!check[i])LCA_Tarjan(i,-1);
	for(i=0;i<q;i++)
	{
		printf("%d\n",d[query[i].u]+d[query[i].v]-2*d[ans[i]]);
	}
}
void read()
{
	int i;
	Enum=0;
	memset(head,-1,sizeof(head));
	for(i=0;i<m;i++)
	{
		int u,v,w,s[5];
		scanf("%d%d%d%s",&u,&v,&w,s);
		insert(u,v,w);
	}
	for(i=1;i<=n;i++)
		node[i].clear();
	scanf("%d",&q);
	for(i=0;i<q;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		query[i].u=u;
		query[i].v=v;
		query[i].num=i;
		
		Node cnt;
		cnt.num=i;
		
		cnt.v=v;
		node[u].push_back(cnt);
		
		cnt.v=u;
		node[v].push_back(cnt);
	}
}
int main()
{
 	while(scanf("%d%d",&n,&m)!=EOF)
 	{
 		read();
 		function();
 	}   
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值