hdu-1598 并查集+贪心

最新推荐文章于 2020-08-03 15:47:22 发布

原创最新推荐文章于 2020-08-03 15:47:22 发布 · 913 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#merge #struct

本文详细介绍了解决 HDU 1598 问题的算法思路，通过使用最小生成树算法寻找两个指定节点间的最小代价路径。文章包括完整的 C 语言代码实现，并解释了关键函数如 find 和 merge 的作用。

可能是太热了，效率低，GDBing很长时间，并贡献WA一个。

/*
 * hdu-1598
 * mike-w
 * 2011-11-10
 */
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define MAXM 1024
#define MAXN 256
#define INF 999999

typedef struct _edge
{
	int start,end,weight;
}edge;

int N,M,Q,S,T;
edge road[MAXM];
int set[MAXN];

int comp(const void *e1, const void *e2)
{
	return ((edge*)e1)->weight - ((edge*)e2)->weight;
}

int find(int e)
{
	int node=e,root;
	while(set[node]>0)
		node=set[node];
	root=node;
	while(set[e]>0)
	{
		node=set[e];
		set[e]=root;
		e=node;
	}
	return root;
}

int merge(int e1,int e2)
{
	int r1=find(e1);
	int r2=find(e2);
	if(r1==r2) return 0;
	if(set[r1]<set[r2])
		set[r2]=r1;
	else if(set[r2]<set[r1])
		set[r1]=r2;
	else
		set[r1]=r2,set[r2]--;
	return 1;
}

int main(void)
{
	int i,j,delta,ans;

#ifndef ONLINE_JUDGE
	freopen("in","r",stdin);
#endif

	while(scanf("%d%d",&N,&M)!=EOF)
	{
		for(i=1;i<=M;i++)
			scanf("%d%d%d",
					&(road[i].start),&(road[i].end),&(road[i].weight));
		scanf("%d",&Q);
		qsort(road+1,M,sizeof(edge),comp);

		while(Q-->0)
		{
			scanf("%d%d",&S,&T);	
			ans=INF;

			for(i=1;i<=M;i++)
			{
				delta=0;
				for(j=0;j<=N;j++)
					set[j]=-1;

				for(j=i;j<=M && find(S)!=find(T);j++)
				{
					if(merge(road[j].start,road[j].end) && 
							(road[j].weight-road[i].weight>delta))
						delta=road[j].weight-road[i].weight;
				}
				if(find(S)==find(T) && ans>delta)
					ans=delta;
				if(find(S)!=find(T))
					break;
			}
			if(ans==INF)
				puts("-1");
			else
				printf("%d\n",ans);
		}
	}
	return 0;
}