最短路之旅行(Dijkstra解法)

我与影子孤独终老i 提交于 2019-12-13 02:43:49

链接:https://ac.nowcoder.com/acm/problem/14550
来源:牛客网

题目描述

小z放假了,准备到RRR城市旅行,其中这个城市有N个旅游景点。小z时间有限,只能在三个旅行景点进行游玩。小明租了辆车,司机很善良,说咱不计路程,只要你一次性缴费足够,我就带你走遍RRR城。

小z很开心,直接就把钱一次性缴足了。然而小z心机很重,他想选择的路程尽量长。

然而司机也很聪明,他每次从一个点走到另外一个点的时候都走最短路径。

你能帮帮小z吗?

需要保证这三个旅行景点一个作为起点,一个作为中转点一个作为终点。(一共三个景点,并且需要保证这三个景点不能重复).

输入描述:
本题包含多组输入,第一行输入一个整数t,表示测试数据的组数
每组测试数据第一行输入两个数N,M表示RRR城一共有的旅游景点的数量,以及RRR城中有的路的数量。
接下来M行,每行三个数,a,b,c表示从a景点和b景点之间有一条长为c的路
t<=40
3<=N,M<=1000
1<=a,b<=N
1<=c<=100
输出描述:
每组数据输出两行,
每组数据包含一行,输出一个数,表示整条路程的路长。
如果找不到可行解,输出-1.

输入

4
7 7
1 2 100
2 3 100
1 4 4
4 5 6
5 6 10
1 6 4
6 7 8
7 3
1 2 1
1 3 1
1 3 2
7 3
1 2 1
3 4 1
5 6 1
8 9
1 2 1
2 3 1
3 4 1
4 1 1
4 5 1
5 6 1
6 7 1
7 8 1
8 5 1

输出

422
3
-1
9

对这个题,我们首先需要计算两两点之间的最短距离。然后,从A经过B到C相当于B到A+B到C,反之亦然。所以我们只需枚举中间点,然后找到从该点出发到其他点距离的最大的两个值的和,并在中间点的和中拿最大的那一个。也就是说,在所有最短距离中取最大。

#include<iostream>
#include<queue>
#include<string.h>
#include<string>
#include<stdio.h>
#include<math.h>
#include<vector>
#include<algorithm>
#define IOS std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
typedef long long ll;
const int N=1e5+5;
const int M=1e5+5;
const int INF=0x3f3f3f3f;
int n,m;
struct edge{
int nxt,to,va;
}e[N];
int head[N],cnt;
void addedge(int a,int b,int c)//建边
{
    e[++cnt].nxt=head[a];
    e[cnt].to=b;
    e[cnt].va=c;    
    head[a]=cnt;
}
int dis[N];
bool vis[N];
int ans=0;
bool cmp(int a,int b)
{    
	return a>b;
}
int Dijkstra(int s)
{    
      for(int i=1;i<=n;i++)
      {
	      vis[i]=0;dis[i]=INF;       
      }    
      dis[s]=0;
	    priority_queue<pair<int,int> >q;
	    q.push({-dis[s],s});
	    while(!q.empty())    
	    {        
	    	int u=q.top().second;//取最小
	        q.pop();        
	        if(!vis[u]) 
	        {            
	             vis[u]=1;           
	             for(int i=head[u];i;i=e[i].nxt)
	            {                
	            	int v=e[i].to;  
	                if(dis[v]>dis[u]+e[i].va)                
	                {                   
	                 dis[v]=dis[u]+e[i].va;//更新
	                 q.push({-dis[v],v});
	                }           
	            } 
	        }    
	    }   
	    sort(dis+1,dis+n+1,cmp); //降序排序
            int x=0;
            for(int i=1;i<n-1;i++)//dis[n-1]==0,即源点,不必考虑
            {       
             if(dis[i]!=INF&&dis[i+1]!=INF)
                         return dis[i]+dis[i+1];    
             }    
     return -1;
}
int main()
{    
	IOS;
	int t;    
	cin>>t;    
	while(t--)    
	{        
		cin>>n>>m;        
		ans=-1;cnt=0;        
		memset(head,0,sizeof head);        
		for(int i=1;i<=m;i++)        
		{       
		     int a,b,c;            
		     cin>>a>>b>>c;            
		     addedge(a,b,c);            
		     addedge(b,a,c);    
   		 }        
   		 for(int i=1;i<=n;i++)        
   		 {      
   		       ans=max(Dijkstra(i),ans);        
   		 }
   		cout<<ans<<endl;    
   	}
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!