PTA L2-001紧急救援

https://pintia.cn/problem-sets/994805046380707840/problems/994805073643683840

最短路问题 求最短路的同时还希望最后的人数最多 还要保存路径

题解:一切尽在代码中

#include<cstdio>
#include<cstring>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<stdlib.h>
#include<stack> 
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=1000000;
int dis[maxn];//最短路 
int tis[maxn];//最多人 
int e[1000][1000];
int vis[1000];
int p[maxn];//记录路径 
int sum[maxn];//记录路数 
int a[maxn];
int n,m,s,d;
void dij(int s){
	for(int i=0;i<n;i++){
		dis[i]=e[s][i];
		tis[i]=a[s]+a[i];
		vis[i]=0;
		sum[i]=1; 
	}
	dis[s]=0;
	tis[s]=a[s];
	vis[s]=1;
	for(int i=0;i<n-1;i++){
		int minn=inf,u;
		for(int j=0;j<n;j++){
			if(!vis[j]&&dis[j]<minn){
				u=j;
				minn=dis[j];
			}
		}
		vis[u]=1;
		for(int v=0;v<n;v++){
			if(!vis[v]){
				if(dis[v]>dis[u]+e[u][v]){//找最短路 
					dis[v]=dis[u]+e[u][v];
					tis[v]=a[v]+tis[u];
					p[v]=u;
					sum[v]=sum[u];//该点的路数等于上个点的路数 
				}
				else if(dis[v]==dis[u]+e[u][v]&&u!=v){//及说明该U点已经被另一个点松弛过了且松弛后与此时的v点相同 
					sum[v]+=sum[u];//更新路数  注意是 +上 上个点的路数 用来更新到目的点 
				    if(tis[v]<a[v]+tis[u]){//注意 是该点的人数+上上一个点的人数 
				     	tis[v]=a[v]+tis[u];
				     	p[v]=u;
					}
				}			
			}	
		}
	}
}
int main(){
	cin>>n>>m>>s>>d;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			if(i!=j)  e[i][j]=inf;
			else e[i][j]=0;
		}
	}
	while(m--){
		int x,y,z;
		cin>>x>>y>>z;
		if(e[x][y]>z){
			e[x][y]=z;
			e[y][x]=z;
		}
	}
	dij(s);
	cout<<sum[d]<<" "<<tis[d]<<endl;
    int cnt=0;   //回溯数组 
	int repath[1000]; 
    repath[cnt++]=d;
		while(p[d]!=s){
			d=p[d];
			repath[cnt++]=d;
		} 
		repath[cnt++]=s;
		
		while(--cnt)
			cout<<repath[cnt]<<" ";
	    cout<<repath[cnt]<<endl;	
	return 0;
} 

猜你喜欢

转载自blog.csdn.net/gml1999/article/details/88854555