CodeVS 1021 玛丽卡——最短路问题

题目描述

麦克找了个新女朋友，玛丽卡对他非常恼火并伺机报复。
因为她和他们不住在同一个城市，因此她开始准备她的长途旅行。
在这个国家中每两个城市之间最多只有一条路相通，并且我们知道从一个城市到另一个城市路上所需花费的时间。
麦克在车中无意中听到有一条路正在维修，并且那儿正堵车，但没听清楚到底是哪一条路。无论哪一条路正在维修，从玛丽卡所在的城市都能到达麦克所在的城市。
玛丽卡将只从不堵车的路上通过，并且她将按最短路线行车。麦克希望知道在最糟糕的情况下玛丽卡到达他所在的城市需要多长时间，这样他就能保证他的女朋友离开该城市足够远。
编写程序，帮助麦克找出玛丽卡按最短路线通过不堵车道路到达他所在城市所需的最长时间(用分钟表示)。

输入描述

第一行有两个用空格隔开的数N和M，分别表示城市的数量以及城市间道路的数量。1≤N≤1000，1≤M≤N*(N-1)/2。城市用数字1至N标识，麦克在城市1中，玛丽卡在城市N中。
接下来的M行中每行包含三个用空格隔开的数A，B和V。其中1≤A，B≤N，1≤V≤1000。这些数字表示在A和城市B中间有一条双行道，并且在V分钟内是就能通过。

输出描述

输出文件的第一行中写出用分钟表示的最长时间，在这段时间中，无论哪条路在堵车，玛丽卡应该能够到达麦克处，如果少于这个时间的话，则必定存在一条路，该条路一旦堵车，玛丽卡就不能够赶到麦克处。

样例输入

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

样例输出

27

题解

一道最短路的模板题,可惜TLE好久好久，从Floyd换到了Dijkstra算法，总有几个数据点超时。渣成狗了。。。。
正确思路：先求最短路，并记录起点到终点的最短路路径。然后以此假设最短路上的路径堵车，再求每种情况的最短路。在所有的最短路里面求最大值。
错误思路：依次枚举每条边堵车的情况，求最短路。。。（1000个点，最多1000*(1000 - 1) / 2 条边，等于求了499500次最短路。。不超时是不是就见鬼了。。。。）

AC代码：

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<limits.h>
using namespace std;
const int N = 1005;
typedef struct Edge{
    int from , to , value;
}edge;
vector<edge> v[N];
int n , m;
bool visit[N][N];//将堵车的路状态置为true
int e[N][N];//存边
int dis[N];//求最短路的状态数组
bool vis[N];//将已经求得最短路的点状态置为true
int path[N];//用来记录最短路的路径

//dijkstra算法:start-->最短路起点，step-->判断是否是第一次求最短路（第一次求最短路不存在堵车的情况）
int dijkstra(int start , int step){
    dis[start] = 0;
    int cnt = 0;
    path[1] = 0;
    while(cnt < n){
        int temp = INT_MAX;
        int tempIndex;
        for(int i = 1;i <= n;i++){
            if(!vis[i] && temp > dis[i]){
                temp = dis[i];
                tempIndex = i;
            }
        }
        vis[tempIndex] = true;
        cnt++;
        for(int i = 1;i <= n;i++){
            if(!vis[i] && e[tempIndex][i] != -1 &&!visit[tempIndex][i] && dis[i] > dis[tempIndex] + e[tempIndex][i]){   
                dis[i] = dis[tempIndex] + e[tempIndex][i];
                visit[tempIndex][i] = true;
                if(step == 1){
                    path[i] = tempIndex;
                }       
            }
        }
    }
    return dis[n];
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> n >> m;
    int temp;
    int from , to ;
    for(int i = 1;i <= n;i++){
        for(int j = 1;j <= n;j++){
            e[i][j] = -1;
        }
    }
    for(int i = 0;i < m;i++){
        cin >> from >> to >> temp;
        e[to][from] = e[from][to] = temp;
    }
    int ans = 0;
    memset(vis , false , sizeof(vis));
    memset(visit , false , sizeof(visit));
    for(int i = 1;i <= n;i++){
        dis[i] = INT_MAX;
    }
    dijkstra(1 , 1);
    int t = n;
    while(t != 1){
        memset(vis , false , sizeof(vis));
        memset(visit , false , sizeof(visit));
        visit[t][path[t]] = true;
        visit[path[t]][t] = true;
        for(int i = 1;i <= n;i++){
            dis[i] = INT_MAX;
        }
        temp = dijkstra(1 , -1);
        ans = max(ans , temp);
        t = path[t];
    }
    cout << ans << endl;
    return 0;
}