开始用的folyd显然1000的三次方超时了。
那么只有使用dijkstra来优化,但是题目中要求的是到达vis并且返回原来的位置,所以这里可以使用一个二维的数组来代表所有顶点到其他点的最小距离。
本题将dijkstra的使用范围和floyd一样,但是复杂度是o(e*logv),比floyd更快
// 求解从顶点s出发到所有点的最短距离
void dijkstra(int s)
{
priority_queue <P,vector<P>,greater<P> > que;
memset(d[s], 0x3f, maxn * sizeof(int));
d[s][s] = 0;
que.push(P(0,s));
while(!que.empty())
{
P p=que.top();que.pop();
int v=p.second;
if(d[s][v]<p.first) continue;
for(int i=0;i<g[v].size();i++)
{
edge e=g[v][i];
if(d[s][e.to]>d[s][v]+e.cost)
{
d[s][e.to]=d[s][v]+e.cost;
que.push(P(d[s][e.to],e.to));
}
}
}
}
*****************************************************************************************************
代码:
#include <iostream>
#include <cstdio>
#include <queue>
#include <string.h>
using namespace std;
const int maxn = 1005;
const int INF= 10000000;
int v,e,vis;
struct edge
{
int to,cost;
};
typedef pair <int ,int >P;
vector<edge> g[maxn];
int d[maxn][maxn];
void dijkstra(int s)
{
priority_queue <P,vector<P>,greater<P> > que;
memset(d[s], 0x3f, maxn * sizeof(int));
d[s][s] = 0;
que.push(P(0,s));
while(!que.empty())
{
P p=que.top();que.pop();
int v=p.second;
if(d[s][v]<p.first) continue;
for(int i=0;i<g[v].size();i++)
{
edge e=g[v][i];
if(d[s][e.to]>d[s][v]+e.cost)
{
d[s][e.to]=d[s][v]+e.cost;
que.push(P(d[s][e.to],e.to));
}
}
}
}
int main ()
{
scanf("%d%d%d",&v,&e,&vis);
--vis;
for(int i=0;i<e;i++)
{
int from,to,cost;
scanf("%d%d%d",&from,&to,&cost);
from--;
to--;
struct edge e;
e.to=to;
e.cost=cost;
g[from].push_back(e);
}
for(int i=0;i<v;i++)
{
dijkstra(i);
}
int ans = 0;
for (int i = 0; i < v; ++i)
{
if (i == vis)
{
continue;
}
ans = max(ans, d[i][vis] + d[vis][i]);
}
printf("%d\n",ans);
}