牛客——旅行

链接：旅行 (nowcoder.com)
来源：牛客网

题目描述

小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.

 该题用Floyd会超时，由于是稀疏图，可以用链式前向星构图，用堆优化的Dijkstra算法求最短路。然后枚举每个点当作中间点，求出由当前点到其他任意两点最长的两个路径。时间复杂度是O(VElogV)。

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N = 1005, INF = 0x3f3f3f3f;
struct node{
    int x, d;
    bool operator< (const node& a) const
    {
        return d > a.d;
    }
};
int dis[N];
int Next[2*N], e[2*N], head[2*N], val[2*N]; //因为一条边会存两次，x->y和y->x各一次, 所以要开两倍大小 
int fg[N] {0}; 
int n, m, cnt = 0;
void add(int x, int y, int z)
{
	e[cnt] = y;
	val[cnt] = z;
	Next[cnt] = head[x];
    head[x] = cnt ++;
}
int Dijkstra(int x)
{
	priority_queue<node> q;
    memset(dis, 0x3f, sizeof(dis));
    memset(fg, 0, sizeof(fg));
    dis[x] = 0;
    q.push({x, 0});
    while(!q.empty())
    {
        node k = q.top();
        q.pop();
        if(fg[k.x]) continue;
        fg[k.x] = 1;
        for(int i=head[k.x]; i!=-1; i=Next[i])
        {
        	if(dis[e[i]]>dis[k.x]+val[i])
        	{
        		dis[e[i]] = dis[k.x] + val[i];
        		q.push({e[i], dis[e[i]]});
			}
		}
    }
    sort(dis+1, dis+n+1, greater<int>());
    int kk = 0, sum = 0;
    for(int i=1; i<n && kk<2; i++) //找到最远的两条边
		if(dis[i]!=INF) sum += dis[i], kk ++;

    return (kk<2)? -1:sum;//当kk<2时表示x不能到达其他任意两个点，返回-1 
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int t;
    cin >> t;
    while(t --)
    {
    	int maxv = -1;
    	cnt = 0;
    	memset(head, -1, sizeof(head));
        for(int i=0; i<=n; i++) e[i] = val[i] = Next[i] = 0;
    	cin >> n >> m;
    	for(int i=0; i<m; i++)
    	{
    		int x, y, z;
    		cin >> x >> y >> z;
    		add(x, y, z);
            add(y, x, z);
		}
		for(int i=1; i<=n; i++) maxv = max(maxv, Dijkstra(i)); //逐一枚举中转点 
		cout << maxv << endl;
	}
	return 0;
}