题目大意:给出n和m表示顶点数和边数,边为双向边,接下来m行每行输入三个数起点,终点,权值。问删除掉其中的一条边,剩下的m-1条边中1到n的最短路中的最大长度是多少。
思路:如果每条边都删除一次尝试着求m次最短路算出最大值肯定会超时,因为边最多有500000条,那么我们可以换个思路,先求出一遍最短路,如果删除的边不是最短路里的边,那么删除这条边后1到n的最短路还是不变的,如果要改变那么删除的边肯定在最短路里。这样子循环最多只有n-1次,大大减少了时间。我们可以在求最短路时记录一下当前所删的边是哪条,记录一下边的起点和终点,当使用SPFA求最短路时发现这条边那么跳过这条边即可。
记录路径以及还原路径递归版和vector版都在下面,递归版本比较耗时间。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<cmath>
#include<map>
#include<string>
#include<vector>
#define inf 0x3fffffff
using namespace std;
const int maxn=1050;
int first[maxn],pre[maxn],sign;
struct node
{
int to,w,next;
}edge[maxn*maxn];
void creat()
{
for(int i=0;i<maxn;i++)
first[i]=0;
sign=1;
}
void add_edge(int u,int v,int w)
{
edge[sign].to=v;
edge[sign].w=w;
edge[sign].next=first[u];
first[u]=sign++;
}
int tmpu,tmpv;
int SPFA(int n)
{
memset(pre,-1,sizeof(pre));
queue<int>q;
int dis[maxn],vis[maxn],p=1;
for(int i=0;i<maxn;i++)
{
dis[i]=inf;
vis[i]=0;
}
q.push(p);
dis[p]=0;
while(!q.empty())
{
int top=q.front();
q.pop();
vis[top]=0;
for(int i=first[top];i;i=edge[i].next)
{
int to=edge[i].to;
int w=edge[i].w;
if((tmpu==top&&tmpv==to)||(tmpu==to&&tmpv==top))continue;
if(dis[to]>dis[top]+w)
{
dis[to]=dis[top]+w;
pre[to]=top;
if(!vis[to])
{
vis[to]=1;
q.push(to);
}
}
}
}
return dis[n]==inf?-1:dis[n];
}
int path[maxn],sum=0;
/*
void findpath(int num) //递归版找路径
{
if(pre[num]==-1)
{
path[sum++]=num;
return ;
}
findpath(pre[num]);
path[sum++]=num;
}*/
vector<int> get_path(int t) //vector求路径
{
vector<int>path;
for(;t!=-1;t=pre[t])
path.push_back(t); //不断沿着pre[t]走直到t=s;
reverse(path.begin(),path.end()); //这样得到的是从t到s的顺序,所以要翻转
return path;
}
int main()
{
int n,m;
while(scanf("%d %d",&n,&m)!=EOF)
{
vector<int>path;
int ans=0;
creat();
int u,v,w;
for(int i=0;i<m;i++)
{
scanf("%d %d %d",&u,&v,&w);
add_edge(u,v,w);
add_edge(v,u,w);
}
ans=SPFA(n);
//findpath(n);
path=get_path(n);
int from=1;
for(int i=1;i<path.size();i++)
{
int to=path[i];
tmpu=from,tmpv=to;
int flag=SPFA(n);
if(flag!=-1)
ans=max(ans,flag);
from=to;
}
printf("%d\n",ans);
}
return 0;
}