1649 齐头并进（最短路dijkstra)

1649 齐头并进（最短路dijkstra)

在一个叫奥斯汀的城市，有n个小镇（从1到n编号），这些小镇通过m条双向火车铁轨相连。当然某些小镇之间也有公路相连。为了保证每两个小镇之间的人可以方便的相互访问，市长就在那些没有铁轨直接相连的小镇之间建造了公路。在两个直接通过公路或者铁路相连的小镇之间移动，要花费一个小时的时间。
现在有一辆火车和一辆汽车同时从小镇1出发。他们都要前往小镇n，但是他们中途不能同时停在同一个小镇（但是可以同时停在小镇n）。火车只能走铁路，汽车只能走公路。
现在请来为火车和汽车分别设计一条线路；所有的公路或者铁路可以被多次使用。使得火车和汽车尽可能快的到达小镇n。即要求他们中最后到达小镇n的时间要最短。输出这个最短时间。(最后火车和汽车可以同时到达小镇n，也可以先后到达。)
样例解释：
在样例中，火车可以按照

1⟶3⟶4
行驶，汽车

1⟶2⟶4
按照行驶，经过2小时后他们同时到过小镇4。
输入
单组测试数据。
第一行有两个整数n 和 m (2≤n≤400, 0≤m≤n*(n-1)/2) ，表示小镇的数目和铁轨的数目。
接下来m行，每行有两个整数u 和 v，表示u和v之间有一条铁路。(1≤u,v≤n, u≠v)。
输入中保证两个小镇之间最多有一条铁路直接相连。
输出
输出一个整数，表示答案，如果没有合法的路线规划，输出-1。
输入样例

4 2
1 3
3 4
输出样例
2
也可以只开一个，因为肯定不会重复

#include<bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
int n,m;
bool vis[405];
int dis[405],train[405][405],bus[405][405],u,v;
void dij(int map[][405])
{memset(dis, 0x3f, sizeof(dis));memset(vis, 0, sizeof(vis));dis[1]=0;for(int i=1;i<=n;i++){int tmp=INF,date=-1;for(int j=1;j<=n;j++)if(!vis[j]&&tmp>dis[j]){tmp=dis[j];date=j;}if (tmp==INF)break;vis[date]=true;for(int j=1;j<=n;j++)if(!vis[j]&&dis[j]>dis[date]+map[date][j])dis[j] = dis[date] + map[date][j];}
}
int main()
{cin>>n>>m;memset(train,0x3f3f3f3f,sizeof(train));memset(bus,0x3f3f3f3f,sizeof(bus)); for (int i=0;i<m;i++){cin>>u>>v;train[u][v]=train[v][u]=1;}for(int i=1;i<=n;i++)for (int j=i+1;j<=n;j++)if (train[i][j]==INF)bus[i][j]=bus[j][i]=1;dij(train);int ans1=dis[n];dij(bus);int ans2=dis[n];int ans=max(ans1,ans2);printf("%dn",ans==INF?-1:ans);return 0;
}