1087 All Roads Lead to Rome (30 分)

最新推荐文章于 2024-09-12 19:02:20 发布

vs5

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量150

点赞数 1

分类专栏： PAT C++ 文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_55591643/article/details/122926607

版权

C++ 同时被 2 个专栏收录

50 篇文章 1 订阅

订阅专栏

PAT

44 篇文章 0 订阅

订阅专栏

多条件限制的最短路，数据范围很小，直接写个朴素版。

用两个map存名字对应编号，和编号对应名字

见代码

#include <iostream>
#include <unordered_map>
#include <cstring>
using namespace std;
const int N = 300;
int g[N][N],dis[N],disha[N],ha[N],path[N],cnt[N],res[N],n,m;
bool st[N];
string start,en = "ROM";
unordered_map<string,int>mp;
unordered_map<int,string>idx;
void dijkstra()
{
	memset(dis,0x3f,sizeof dis);
	dis[mp[start]] = 0;
	res[mp[start]] = 1;
	for(int i = 0; i < n; i ++)
	{
		int u = -1;
		for(int j = 1; j <= n; j ++)
		{
			if(!st[j] && (u == -1 || dis[u] > dis[j])) u = j;
		}  
		st[u] = true;
		for(int j = 1; j <= n; j ++)
		{
			if(!st[j] && dis[j] > dis[u] + g[u][j])
			{
				dis[j] = dis[u] + g[u][j];// 最短路 
				disha[j] =disha[u] + ha[j];//幸福 
				cnt[j] = cnt[u] + 1;// 路径点数 
				res[j] = res[u];//最短路个数 
				path[j] = u;//路径 
			}
			else if(dis[j] == dis[u] + g[u][j])
			{   
				res[j] += res[u];
				if(disha[j] < disha[u] + ha[j])
				{
					disha[j] = disha[u] + ha[j];
					cnt[j] = cnt[u] + 1;
 					path[j] = u;
				}
				else if(disha[j] == disha[u] + ha[j])
					if(cnt[j] > cnt[u] + 1)
					{
						cnt[j] = cnt[u] + 1;
						path[j] = u;
					}
			}
		}
	}
}
void dfs(int u)
{
	if(u == mp[start])
	{
		cout << idx[u];return ;
	}
	dfs(path[u]);
	cout << "->" << idx[u];
}
int main()
{
	cin >> n >> m >> start;
	memset(g,0x3f,sizeof g);
	mp[start] = 1;idx[1] = start;
	for(int i = 2; i <= n ; i ++)
	{
		string s;int x;
		cin >> s >> x;
		mp[s] = i;idx[i] = s;ha[i] = x;
	}
	for(int i = 0; i < m; i ++)
	{
		string a,b;int x;
		cin >> a >> b >> x;
		g[mp[a]][mp[b]] = g[mp[b]][mp[a]] = x;
	}
	dijkstra();
	cout << res[mp[en]] << ' ' << dis[mp[en]] << ' ' << disha[mp[en]] << ' ' << disha[mp[en]] / (cnt[mp[en]]) << endl;
	dfs(mp[en]);
	return 0;
 }