POJ 2387 Til the Cows Come Home(Dijkstra模板题)

题目链接:http://poj.org/problem?id=2387

题意是输入n条边,m个城镇,然后输入a城镇到b城镇的距离,最后问从n到1城镇的最短距离为多少。

最短路的入门题吧,直接用迪杰斯特拉的模板就行了,需要注意的是有一个判重,这个代码中注释的有,看一下就明白了。


AC代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 2005;           //点的个数
const int INF = 0x3f3f3f3f;
int MAP[MAXN][MAXN];             //地图
int n,m;                         //n个点m条边

void init(){                     //初始化
	for(int i=1;i<=m;i++){
		for(int j=1;j<=m;j++){
			if(i == j){
				MAP[i][j] = 0;           //自己到自己的距离为0
			}
			else{
				MAP[i][j] = INF;         //其余的都设置为最大值 意为断路
			}
		}
	}
}

void dijkstra(){
	int Min,flag;                  //一个记录最小值一个标记位置
	int dis[MAXN];                 //存距离
	bool vis[MAXN];                //标记
	for(int i=1;i<=m;i++){         //初始化
		vis[i] = 0;
		dis[i] = MAP[1][i];
	}
	for(int i=1;i<=m;i++){
		Min = INF;
		for(int j=1;j<=m;j++){       //找出最短边
			if(!vis[j] && dis[j] < Min){
				Min = dis[j];
				flag = j;
			}
	}
		vis[flag] = 1;
		for(int j=1;j<=m;j++){        //松弛
			if(!vis[j] && dis[j] > dis[flag] + MAP[flag][j]){
				dis[j] = dis[flag] + MAP[flag][j];
			}
		}
	}
		printf("%d\n",dis[m]);
}

int main()
{
	while(~scanf("%d%d",&n,&m)){
		init();
		int u,v,w;
		for(int i=1;i<=n;i++){                //加边
			scanf("%d%d%d",&u,&v,&w);
			if(MAP[u][v] > w){              //这道题需要判断是否是重边  比如1 2 20 和 1 2 30 肯定取前者
				MAP[u][v] = MAP[v][u] = w;      //因为MAP初始化为INF 所以不影响第一次加边时的操作
			}
		}
		dijkstra();
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/Charles_Zaqdt/article/details/81220979
今日推荐