畅通工程续(djk+优先队列)

#include <stdio.h>
#include <queue>
#include <string.h>
using namespace std;
typedef pair<int,int>pii;
#define N 100011
#define M 1000011
#define INF 0x3f3f3f3f
struct Edge{//边的定义 
	int to,val,next;
	Edge(){}
	Edge(int _to,int _val,int _next){
		to=_to;val=_val;next=_next;
	}
	
}edge[M<<1]; //如果是双向图的话,边的数量是题目中描述的2倍 
int n,m;//n是图中的点数,m 是图中的边数 
int head[N],top;
void init(int n){//初始化 
	memset(head,-1,sizeof(int)*(n+1));//把从head开始的n+1个int类型值设置为-1通常用来初始化数组
	top=0;
}
void add(int u,int v,int val){//加单向边 
	edge[top]=Edge(v,val,head[u]);//head[u]与u同起点上一条边的位置 
	head[u]=top++;//把top的位置赋值给head[u],便于下一次接着使用 
}
void getmap(int m)//建图 
{
	int u,v,val;
	while(m--)
	{
		scanf("%d%d%d",&u,&v,&val);
		add(u,v,val);
		add(v,u,val);//如果是双向图加上这个代码 
	}
}
int dis[N];
void djk(int st,int ed)
{
	memset(dis,0x3f,sizeof(int)*(n+1));
	priority_queue<pii,vector<pii>,greater<pii> > que;//make_pair(dis,v)表示v到起点的距离为dis 
	dis[st]=0;que.push(make_pair(0,st));
	while(!que.empty()) {
		pii p=que.top();que.pop();//每次取离起点最近的 
		int v=p.second;
		if(dis[v]<p.first) continue;
		for(int i=head[v];~i;i=edge[i].next)
		{
			Edge e=edge[i];
			if(dis[e.to]>dis[v]+e.val){
				dis[e.to]=dis[v]+e.val;
				que.push(make_pair(dis[e.to],e.to));
			}
		}
	}
	printf("%d\n",dis[ed]==INF?-1:dis[ed]);
}
int main()
{
	int t,a,b;
	while(scanf("%d%d",&n,&t)!=EOF)
	{
	init(n);
	getmap(t);
	scanf("%d%d",&a,&b);
	djk(a,b);
	}
	return 0;
}

某省自从实行了很多年的畅通工程计划后,终于修建了很多路。不过路多了也不好,每次要从一个城镇到另一个城镇时,都有许多种道路方案可以选择,而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。 

现在,已知起点和终点,请你计算出要从起点到终点,最短需要行走多少距离。

Input

本题目包含多组数据,请处理到文件结束。 
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000),分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0~N-1编号。 
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。 
再接下一行有两个整数S,T(0<=S,T<N),分别代表起点和终点。

Output

对于每组数据,请在一行里输出最短需要行走的距离。如果不存在从S到T的路线,就输出-1. 

Sample Input

3 3
0 1 1
0 2 3
1 2 1
0 2
3 1
0 1 1
1 2

Sample Output

2
-1

猜你喜欢

转载自blog.csdn.net/qq_42434171/article/details/81675799