1003. Emergency (25)

最新推荐文章于 2020-02-17 11:42:09 发布

6点A君

最新推荐文章于 2020-02-17 11:42:09 发布

阅读量593

点赞数

CC 4.0 BY-SA版权

分类专栏： pat a 文章标签： acm c语言 pat

本文链接：https://blog.youkuaiyun.com/anLA_/article/details/53997586

pat a 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一道PAT竞赛题目，旨在寻找从当前城市到发生火灾的目标城市的最短路径及途中能集结的最大消防员数量。文章详细解释了如何使用Dijkstra算法解决此问题，并提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pat

这题大概意思是，众多城市点，然后某一个城市发生火灾，求你所在城市到目标城市的最短路径，并且能够赶到目标城市的最多消防员，

也就是最短路径上涉及所有的消防员之和。

解题心得：

1、Dijkstra算法，就是求源点到目标点最短路径。这个算法有点忘，看了书，后来才回想起来。

2、dijkstra算法的意思，就是把点分为两个集合，一个集合是已经算出最短路径，另一个就是待算的，

每次从已经求到的最短路径集合里面找出最短，再去和未知集合比较，从而更改源点到目标点的距离，

将求得的点标记好，放入到已经求成功的集合。

3、map[myStart][myEnd] = map[myEnd][myStart] = myLength; 就是这一句，一开始理解为有向图，

以为只有一个方向的，然后没有把后者加上。

代码如下：

#include"stdio.h"
#include"string.h"
#include"math.h"
#include<iostream>
#include<vector>
using namespace std;
#define INF 0x7FFFFFFF
struct city{
	int roadNum;
	int emergencyNum;
	int distance;
};
vector<int> team;
vector<bool> visit;
vector<vector<int>> map;
vector<struct city> cities;

int countShortest(){
	int min = INF;
	int index = -1;
	for(int i = 0;i < cities.size();i++){
		if(visit[i] == false && cities[i].distance < min){
			min = cities[i].distance;
			index = i;
		}
	}
	return index;
}

void digistra(int start,int end){
	//initialization
	for(int i = 0;i < cities.size();i++){
		cities[i].distance = INF;
		cities[i].emergencyNum = 0;
		cities[i].roadNum = 0;
	}
	cities[start].distance = 0;
	cities[start].emergencyNum = team[start];
	cities[start].roadNum = 1;

	//monify the shortest ways of each point
	while(true){
		int index = countShortest();
		if(index == -1){
			return;
		}
		visit[index] = true;
		if( index == end ){
			return;
		}
		for(int i = 0;i < cities.size();i++){
			if(visit[i] || map[index][i] == INF){
				continue;
			}
			//modify
			if(cities[i].distance > cities[index].distance + map[index][i]){
				cities[i].distance =  cities[index].distance + map[index][i];
				cities[i].emergencyNum = cities[index].emergencyNum + team[i];
				cities[i].roadNum = cities[index].roadNum;
			}else if(cities[i].distance == cities[index].distance + map[index][i]){
				cities[i].roadNum += cities[index].roadNum;
				if(cities[i].emergencyNum < cities[index].emergencyNum + team[i]){
					cities[i].emergencyNum = cities[index].emergencyNum + team[i];
				}
			}
		}

	}
}

int main(){
	int cityNum;
	while(scanf("%d",&cityNum) != EOF){
		int roadNum,startPoint,endPoint;
		scanf("%d%d%d",&roadNum,&startPoint,&endPoint);
		team.resize(cityNum);

		visit.resize(cityNum);
		visit.assign(cityNum,false);
		
		map.resize(cityNum);

		for(int i = 0;i < cityNum;i++)
		map[i].assign(cityNum,INF);
		
		cities.resize(cityNum);

		for(int i = 0;i < cityNum;i++){
			scanf("%d",&team[i]);
		}

		for(int i = 0;i < roadNum;i++){
			int myStart,myEnd,myLength;
			scanf("%d%d%d",&myStart,&myEnd,&myLength);
			if(map[myStart][myEnd] > myLength){     //get the shortest
				map[myStart][myEnd] = map[myEnd][myStart] = myLength;
			}
		}
		digistra(startPoint,endPoint);
		printf("%d %d",cities[endPoint].roadNum,cities[endPoint].emergencyNum);
	}
	return 0;
}