1. 程式人生 > >hdu2066一個人的旅行解題報告---Floyd演算法稍加優化

hdu2066一個人的旅行解題報告---Floyd演算法稍加優化

一個人的旅行

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 52364    Accepted Submission(s): 17422

 

Problem Description

雖然草兒是個路痴(就是在杭電待了一年多,居然還會在校園裡迷路的人,汗~),但是草兒仍然很喜歡旅行,因為在旅途中 會遇見很多人(白馬王子,^0^),很多事,還能豐富自己的閱歷,還可以看美麗的風景……草兒想去很多地方,她想要去東京鐵塔看夜景,去威尼斯看電影,去陽明山上看海芋,去紐約純粹看雪景,去巴黎喝咖啡寫信,去北京探望孟姜女……眼看寒假就快到了,這麼一大段時間,可不能浪費啊,一定要給自己好好的放個假,可是也不能荒廢了訓練啊,所以草兒決定在要在最短的時間去一個自己想去的地方!因為草兒的家在一個小鎮上,沒有火車經過,所以她只能去鄰近的城市坐火車(好可憐啊~)。

Input

輸入資料有多組,每組的第一行是三個整數T,S和D,表示有T條路,和草兒家相鄰的城市的有S個,草兒想去的地方有D個;
接著有T行,每行有三個整數a,b,time,表示a,b城市之間的車程是time小時;(1=<(a,b)<=1000;a,b 之間可能有多條路)
接著的第T+1行有S個數,表示和草兒家相連的城市;
接著的第T+2行有D個數,表示草兒想去地方。

Output

輸出草兒能去某個喜歡的城市的最短時間。

Sample Input

6 2 3

1 3 5

1 4 7

2 8 12

3 8 4

4 9 12

9 10 2

1 2

8 9 10

Sample Output

9

Author

Grass

思路:題目好像Dijkstra也可以寫,把草兒附近的城市點的距離賦值:

path[1][i] = 0 && path[i][1] = 0即可

不過資料不大,加下優化後Floyd也不會超時。

#include<iostream>
#include<sstream>
#include<cstdlib>
#include<cmath>
#include<cctype>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<map>
#include<vector>
#include<stack>
#include<queue>
#include<set>
#include<list>
#define mod 998244353
#define INF 0x3f3f3f3f
#define Min 0xc0c0c0c0
#define mst(a) memset(a,0,sizeof(a))
#define f(i,a,b) for(int i=a;i<b;i++)
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 5;
int path[1005][1005];
int u, v, t;
void init(){
    for(int i = 1; i <= 1004; i++){
        for(int j = 1; j <= 1004; j++){
            i == j?path[i][j] = 0:path[i][j] = path[j][i] = INF;
        }
    }
}
int main(){
    int T, n, m;
    while(scanf("%d%d%d", &T, &n, &m) != EOF){
        init();
        int max_v = -1;
        for(int i = 0; i < T; i++){
            scanf("%d%d%d", &u, &v, &t);
            max_v = max(u, max(v, max_v));
            if(path[u][v] > t){
                path[u][v] = path[v][u] = t;
            }
        }
        bool vis1[1005] = {false};
        bool vis2[1005] = {false};
        for(int i = 0; i < n; i++){
            scanf("%d", &u);
            vis1[u] = true;
        }
        for(int i = 0; i < m; i++){
            scanf("%d", &v);
            vis2[v] = true;
        }
        int mins = INF;
        for(int k = 1; k <= max_v; k++){
            for(int i = 1; i <= max_v; i++){
                if(path[i][k] == INF) continue; //這個判斷很重要
                for(int j = 1; j <= max_v; j++){
                    path[i][j] = min(path[i][k] + path[k][j], path[i][j]);
                    if(vis1[i] && vis2[j]){
                        mins = min(path[i][j], mins);
                    }
                }
            }
        }
        printf("%d\n", mins);
    }
    return 0;
}