HDU 2066 一个人的旅行

一个人的旅行

http://acm.hdu.edu.cn/showproblem.php?pid=2066

分析

• 最短路径
• 一对多，虚拟终点

代码

O(n²)【31MS】

// hdu 2066 一个人的旅行
#include<bits/stdc++.h>
using namespace std;
#define MAXN 1005
#define inf 0x3f3f3f3f
int Map[MAXN][MAXN], vis[MAXN], dis[MAXN];
int main(){
    int T, S, D, a, b, tim, s, t, nxt, Min;
    while(scanf("%d %d %d", &T, &S, &D) == 3){
		memset(vis, 0, sizeof vis);
		memset(dis, inf, sizeof dis);
		memset(Map, inf, sizeof Map);
        for(int i = 0; i < T; i++){
            scanf("%d %d %d", &a, &b, &tim);
            Map[a][b] = min(Map[a][b], tim);
            Map[b][a] = Map[a][b];
        }        
        s = 0, t = 1001;
        while(S--)
            scanf("%d", &a), Map[s][a] = Map[a][s] = 0;
        while(D--)
            scanf("%d", &b), Map[t][b] = Map[b][t] = 0;
        vis[s] = 0, dis[s] = 0;
        while(s != t){
            Min = inf;
            for(int i = 0; i < 1002; i++){
                if(vis[i]) continue;
                if(Map[s][i] != inf)
                    dis[i] = min(dis[i], dis[s]+Map[s][i]);
                if(dis[i] < Min) Min = dis[i], nxt = i;
            }
            if(Min == inf) break; // 路径不存在
            s = nxt, vis[nxt] = 1;
        }
        printf("%d\n", dis[t] == inf ? -1 : dis[t]);
    }
    return 0;
}