L3-007 天梯地图 (30 分)单源最短路 多种优先级

本题要求你实现一个天梯赛专属在线地图,队员输入自己学校所在地和赛场地点后,该地图应该推荐两条路线:一条是最快到达路线;一条是最短距离的路线。题目保证对任意的查询请求,地图上都至少存在一条可达路线。

输入格式:
输入在第一行给出两个正整数N(2 ≤ N ≤ 500)和M,分别为地图中所有标记地点的个数和连接地点的道路条数。随后M行,每行按如下格式给出一条道路的信息:

V1 V2 one-way length time
其中V1和V2是道路的两个端点的编号(从0到N-1);如果该道路是从V1到V2的单行线,则one-way为1,否则为0;length是道路的长度;time是通过该路所需要的时间。最后给出一对起点和终点的编号。

输出格式:
首先按下列格式输出最快到达的时间T和用节点编号表示的路线:

Time = T: 起点 => 节点1 => … => 终点
然后在下一行按下列格式输出最短距离D和用节点编号表示的路线:

Distance = D: 起点 => 节点1 => … => 终点
如果最快到达路线不唯一,则输出几条最快路线中最短的那条,题目保证这条路线是唯一的。而如果最短距离的路线不唯一,则输出途径节点数最少的那条,题目保证这条路线是唯一的。

如果这两条路线是完全一样的,则按下列格式输出:

Time = T; Distance = D: 起点 => 节点1 => … => 终点
输入样例1:
10 15
0 1 0 1 1
8 0 0 1 1
4 8 1 1 1
5 4 0 2 3
5 9 1 1 4
0 6 0 1 1
7 3 1 1 2
8 3 1 1 2
2 5 0 2 2
2 1 1 1 1
1 5 0 1 3
1 4 0 1 1
9 7 1 1 3
3 1 0 2 5
6 3 1 2 1
5 3
输出样例1:
Time = 6: 5 => 4 => 8 => 3
Distance = 3: 5 => 1 => 3
输入样例2:
7 9
0 4 1 1 1
1 6 1 3 1
2 6 1 1 1
2 5 1 2 2
3 0 0 1 1
3 1 1 3 1
3 2 1 2 1
4 5 0 2 2
6 5 1 2 1
3 5
输出样例2:
Time = 3; Distance = 4: 3 => 2 => 5

#include <bits/stdc++.h>
using namespace std;
const int N=505,INF=0x3f3f3f3f,M=250005;

int dist[N],dist2[N],n,m,x,y,flag,l,t,s,d,f[N],ans[N],p,d1,d2,p1;
int idx,h[N],e[M],ne[N],len[M],tim[M];
bool st[N],f2=true;
void add(int a,int b,int c,int d)
{
	e[idx]=b;ne[idx]=h[a];len[idx]=c;tim[idx]=d;h[a]=idx++;
}
class node{
	public :
	int id,tim,len;
	node(){};
	node(int a,int b,int c){
		id=a;tim=b;len=c;
	}
	bool operator <(const node & b)const {
		if(tim!=b.tim)return tim>b.tim;
		return len>b.len;
	} 
};

class node2
{
	public:
		int id,len,num;
		node2(){}
		node2(int a,int b,int c){
			id=a;len=b;num=c;
		}
		bool operator <(const node2 & b)const {
			if(len!=b.len)return len>b.len;
			return num>b.num;
		} 
};

void dijkstra(int u)
{
	memset(dist,0x3f,sizeof dist);
	memset(dist2,0x3f,sizeof dist2);
	priority_queue<node> q;
	q.push(node(u,0,0));
	dist[u]=0;
	dist2[u]=0;
	f[u]=-1;
	while(!q.empty())
	{
		node t=q.top();q.pop();
		if(st[t.id])continue;
		st[t.id]=true;
		for(int i=h[t.id];i!=-1;i=ne[i]){
			int j=e[i];
			if(dist[j]>dist[t.id]+tim[i]){
				dist[j]=dist[t.id]+tim[i];
				dist2[j]=dist2[t.id]+len[i];
				f[j]=t.id;
				q.push(node(j,dist[j],dist2[j]));
			}
			else if(dist[j]==dist[t.id]+tim[i]){
				if(dist2[j]>dist2[t.id]+len[i]) {
					dist2[j]=dist2[t.id]+len[i];
					f[j]=t.id;
					q.push(node(j,dist[j],dist2[j]));
				}
			}
		}
	}
}

void dijkstra2(int u)
{
	memset(dist,0x3f,sizeof dist);
	memset(dist2,0x3f,sizeof dist2);
	memset(st,0,sizeof st);
	dist[u]=0;
	f[u]=-1;
	dist2[u]=1;
	priority_queue<node2> q;
	q.push(node2(u,0,1));
	while(!q.empty()){
		node2 t=q.top();q.pop();
		if(st[t.id])continue;
		st[t.id]=true;
		for(int i=h[t.id];i!=-1;i=ne[i]){
			int j=e[i];
			if(dist[j]>dist[t.id]+len[i]){
				dist[j]=dist[t.id]+len[i];//长度 
				f[j]=t.id;
				dist2[j]=dist2[t.id]+1;//经历的节点的个数 
				q.push(node2(j,dist[j],dist2[j])); 
			}
			else if(dist[j]==dist[t.id]+len[i]){
				if(dist2[j]>dist2[t.id]+1){
					dist2[j]=dist2[t.id]+1;
					f[j]=t.id;
					q.push(node2(j,dist[j],dist2[j])); 
				}
			}
		}
	}
}

void dfs(int u)
{
	if(u==-1)return;
	dfs(f[u]);
	ans[p++]=u;
}

int main()
{
	cin>>n>>m;
	memset(h,-1,sizeof h);
	while(m--)
	{
		cin>>x>>y>>flag>>l>>t;
		add(x,y,l,t);
		if(flag==0)add(y,x,l,t);
	}
	cin>>s>>d;
	dijkstra(s);
	d1=dist[d];
	dfs(d);p1=p;
	dijkstra2(s);
	dfs(d);
	d2=dist[d];
	if(p!=p1*2)flag=false;
	for(int i=0;i<p/2;++i){
		if(ans[i]!=ans[i+p/2])flag=false;
	}
	if(flag){
		printf("Time = %d; Distance = %d: ",d1,d2);
		for(int i=0;i<p/2;++i){
			if(i!=p/2-1)cout<<ans[i]<<" => ";
			else cout<<ans[i];
		}
	}
	else {
		printf("Time = %d: ",d1);
		
		for(int i=0;i<p1;++i){
			if(i!=p1-1)cout<<ans[i]<<" => ";
			else cout<<ans[i];
		}
		cout<<endl;
		printf("Distance = %d: ",d2);
		for(int i=p1;i<p;++i){
			if(i!=p-1)cout<<ans[i]<<" => ";
			else cout<<ans[i];
		}
	}
	return 0;
}
posted @ 2022-11-17 23:02  林动  阅读(9)  评论(0编辑  收藏  举报