hdu 2066：一个人的旅行

草儿决定要在最短的时间去一个自己想去的地方，因为草儿的家在一个小镇上，没有火车经过，所以她只能去邻近的城市坐火车。多组输入数据，每组的第一行是三个整数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个数，表示草儿想去地方。

差不多直接ctrl + c再ctrl + v了。。。

 

一开始看到a，b <= 1000，以为floyd算法可能会超时，不过看到评论区有大神用floyd过了，就写了写。如果没优化的话是妥妥的会TLE的，就象我第一次一样，于是参照别的floyd的AC代码优化了一下，基本上优化在初始化时以对角线i == i划分为两块，这样初始化的时间会减少一半，然后floyd内部可以跳过time[i][k] == inf的情况，时间又会减少不少，这样基本就能过了。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std ;

const int MAXN = 1005 ;
const int inf = 1 << 20 ;
int T , S , D , mm ;
int time[MAXN][MAXN] = {0} ;
bool ss[MAXN] = {0} ;
bool dd[MAXN] = {0} ;

void floyd() {
    int Min = inf ;
    for (int k = 1 ; k <= mm ; k ++) {
        for (int i = 1 ; i <= mm ; i ++) {
            if (time[i][k] != inf) {
                for (int j = 1 ; j <= mm ; j ++) {
                    if (time[i][k] + time[k][j] < time[i][j]) {
                        time[i][j] = time[i][k] + time[k][j] ;
                    }
                    if (ss[i] && dd[j] && time[i][j] < Min) {
                        Min = time[i][j] ;
                    }
                }
            }
        }
    }
    cout << Min << endl ;
}

int main() {
    //freopen("in.txt" , "r" , stdin) ;
    while (cin >> T >> S >> D) {

        mm = 0 ;
        memset(ss , 0 , sizeof(ss)) ;
        memset(dd , 0 , sizeof(dd)) ;
        for (int i = 0 ; i < MAXN ; i ++) {
            for (int j = i + 1 ; j < MAXN ; j ++) {
                time[j][i] = time[i][j] = inf ;
            }
        }

        while (T --) {
            int a , b , tt ;
            scanf("%d%d%d" , &a , &b , &tt) ;
            if (a > mm) mm = a ;
            if (b > mm) mm = b ;
            if (tt < time[a][b]) time[a][b] = time[b][a] = tt ;
        }
        int t ;
        for (int i = 0 ; i < S ; i ++) {
            cin >> t ;
            ss[t] = 1 ;
        }
        for (int i = 0 ; i < D ; i ++) {
            cin >> t ;
            dd[t] = 1 ;
        }
        floyd() ;
    }
    return 0 ;
}