最短路径--Dijstra算法

最短路径


题目描述:

在每年的校赛里,所有进入决赛的同学都会获得一件很漂亮的t-shirt。但是每当我们的工作人员把上百件的衣服从商店运回到赛场的时候,却是非常累的!所以现在他们想要寻找最短的从商店到赛场的路线,你可以帮助他们吗?

输入:

输入包括多组数据。每组数据第一行是两个整数N、M(N<=100,M<=10000),N表示成都的大街上有几个路口,标号为1的路口是商店所在地,标号为N的路口是赛场所在地,M则表示在成都有几条路。N=M=0表示输入结束。接下来M行,每行包括3个整数A,B,C(1<=A,B<=N,1<=C<=1000),表示在路口A与路口B之间有一条路,我们的工作人员需要C分钟的时间走过这条路。输入保证至少存在1条商店到赛场的路线。
当输入为两个0时,输入结束。

输出:

对于每组输入,输出一行,表示工作人员从商店走到赛场的最短时间。

样例输入:
2 1
1 2 3
3 3
1 2 5
2 3 5
3 1 2
0 0
样例输出:
3
2


代码如下:

#include <stdio.h>
#include <vector>

using namespace std;

#define N 101

struct E
{
	int next;
	int cost;
};

vector<E> edge[N];
bool mark[N];

int Dis[N];

int main()
{
	int n, m;
	while (scanf("%d%d", &n, &m) != EOF)
	{
		if (n == 0 && m == 0) break;
		for (int i = 1; i <= n; i++) edge[i].clear();

		while (m--)
		{
			int a, b, c;
			scanf("%d%d%d", &a, &b, &c);
			E tmp;
			tmp.cost = c;
			tmp.next = b;
			edge[a].push_back(tmp);
			tmp.next = a;
			edge[b].push_back(tmp);
		}

		for (int i = 1; i <= n; i++)
		{
			Dis[i] = -1;
			mark[i] = false;
		}
		Dis[1] = 0;
		mark[1] = true;
		int newP = 1;
		for (int i = 1; i < n; i++)
		{
			for (int j = 0; j < edge[newP].size(); j++)
			{
				int t = edge[newP][j].next;
				int c = edge[newP][j].cost;
				if (mark[t] == true) continue;
				if (Dis[t] == -1 || Dis[t]>Dis[newP] + c)
					Dis[t] = Dis[newP] + c;
			}
			int min = 123123123;
			for (int j = 1; j <= n; j++)
			{
				if (mark[j] == true) continue;
				if (Dis[j] == -1) continue;
				if (Dis[j] < min)
				{
					min = Dis[j];
					newP = j;
				}
			}
			mark[newP] = true;
		}
		printf("%d\n", Dis[n]);
	}
	return 0;
}


你可能感兴趣的:(最短路径,九度OJ,dijstra)