pta-L3-011 直捣黄龙 dijkstra

题目

题意：求最短路，同等长度下取经过最多城镇的，同等城镇数下取杀敌最多的。

输出最短路径数，最短距离，杀敌数。

题解：

和L2的紧急救援非常类似。三角更新的时候分类讨论即可。

输出是倒序可以用函数递归输出。

地名因为是三个大写字母所以用26进制表示，然后开数组存一下对应关系就好了。

代码：

#include <iostream>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
const int inf=0x3f3f3f3f;
int s2n(string s){
	return (s[0]-'A')*26*26+(s[1]-'A')*26+s[2]-'A';
}
string n2s(int n){
	char c[3];
	c[2]=n%26+'A';
	n/=26;
	c[1]=n%26+'A';
	n/=26;
	c[0]=n%26+'A';
	string s;
	s.append(1,c[0]);
	s.append(1,c[1]);
	s.append(1,c[2]);
	return s;
}
int mp[30000];//26进制数到点序号
int re_mp[30000];
int cnt=0;
int add(string s){
	if(!mp[s2n(s)]){
		cnt++;
		mp[s2n(s)]=cnt;
		re_mp[cnt]=s2n(s);
		return cnt;
	}
	return mp[s2n(s)];
}
int cost[205][205];
int dis[205];
bool vis[205];
int pre[205];
int enemy[205];
int town_num[205];
int enemy_num[205];
int road_num[205];
void dijk(int st,int n){
	int i,j;
	memset(vis,false,sizeof(vis));
	for(i=1;i<=n;i++)dis[i]=inf;
	dis[st]=0;
	road_num[st]=1;
	for(j=1;j<=n;j++){
		int k=-1;
		for(i=1;i<=n;i++){
			if(!vis[i]&&(k==-1||dis[k]>dis[i])){
				k=i;
			}
		}
		if(k==-1)break;
		vis[k]=true;
		for(i=1;i<=n;i++){
			if(vis[i])continue;
			if(dis[k]+cost[k][i]<dis[i]){
				dis[i]=dis[k]+cost[k][i];
				pre[i]=k;
				town_num[i]=town_num[k]+1;
				enemy_num[i]=enemy_num[k]+enemy[i];
				road_num[i]=road_num[k];
			}
			else if(dis[k]+cost[k][i]==dis[i]){
				road_num[i]+=road_num[k];
				if(town_num[k]+1>town_num[i]){
					town_num[i]=town_num[k]+1;
					enemy_num[i]=enemy_num[k]+enemy[i];
					pre[i]=k;
				}
				else if(town_num[k]+1==town_num[i]){
					if(enemy_num[k]+enemy[i]>enemy_num[i]){
						enemy_num[i]=enemy_num[k]+enemy[i];
						pre[i]=k;
					}
				}
			}
		}
	}
}
void print(int en){
	if(pre[en]==0){
		cout << n2s(re_mp[en]);
		return ;
	}
	print(pre[en]);
	cout<<"->"<<n2s(re_mp[en]);
	return ;
}
int main()
{

	std::ios::sync_with_stdio(false);
	string s;
	int n,m,i,j,a,b,c;
	int st,en;
	cin>>n>>m;
	cin>>s;
	add(s);
	st=add(s);
	cin>>s;
	add(s);
	en=add(s);
	for(i=1;i<=n;i++)for(j=1;j<=n;j++)cost[i][j]=inf;
	for(i=1;i<n;i++){
		cin>>s;
		a=add(s);
		cin>>b;
		enemy[a]=b;
	}
	for(i=1;i<=m;i++){
		cin>>s;
		a=add(s);
		cin>>s;
		b=add(s);
		cin>>c;
		cost[a][b]=cost[b][a]=c;
	}
	dijk(st,n);
	print(en);
	cout<< endl;
	cout <<road_num[en]<<" "<<dis[en]<<" "<<enemy_num[en]<<endl;
	/*for(i=1;i<=n;i++){
		for(j=1;j<=n;j++){
			cout<<i<<" "<<j<<':'<<cost[i][j]<<" ";
		}
		cout<<endl;
	}*/
    //cout << "Hello world!" << endl;
    return 0;
}