hdu 2066 一个人的旅行 解题报告

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstdlib>
 4 #include <cstring>
 5 using namespace std;
 6 
 7 const int INF = 1 << 20;  // INF 的设置一定要足够的大
 8 const int maxn = 1000 + 5;
 9 int dist[maxn], map[maxn][maxn], vis[maxn];
10 int T, S, D, n;
11 
12 void Init()
13 {
14     for (int i = 0; i < maxn; i++)  // 只能小于，不能等于，否则会TLE!!
15     {
16         for (int j = 0; j < maxn; j++)  // 只能小于，不能等于, 否则会TLE!!
17             map[i][j] = (i == j ? 0 : INF);
18     }
19     int st, end, time;
20     n = 0;
21     for (int i = 0; i < T; i++)
22     {
23         scanf("%d%d%d", &st, &end, &time);
24         if (map[st][end] > time)  // 有重边
25             map[st][end] = map[end][st] = time;
26         // 找出最大边的编号
27         n = max(n, max(st, end));
28     }
29     n++;   // 假设是理想中的终点（比最大的顶点大1）的编号
30     for (int i = 0; i < S; i++)
31     {
32         scanf("%d", &st);
33         map[0][st] = map[st][0] = 0; // 草儿家到相邻城市的距离为0
34     }
35     for (int i = 0; i < D; i++)
36     {
37         scanf("%d", &st);
38         map[st][n] = map[n][st] = 0;  // 想去的地方到理想中的终点距离为0
39     }
40 }
41 
42 void Dijkstra()
43 {
44     for (int i = 0; i <= n; i++)  // 从0改为1
45         dist[i] = map[0][i];  // 以草儿家和她相邻的点作为起点，求出该起点到相邻点的时间
46     memset(vis, 0, sizeof(vis));
47     for (int i = 0; i <= n; i++)    // 0 改为 1 也行
48     {
49         int u;
50         int maxx = INF;
51         for (int j = 0; j <= n; j++)  
52         {
53             if (!vis[j] && dist[j] < maxx)
54                 maxx = dist[u=j];
55         }
56         vis[u] = 1;
57         for (int j = 0; j <= n; j++)
58         {
59             if (dist[j] > dist[u] + map[u][j])
60                  dist[j] = dist[u] + map[u][j];
61         }
62     }
63 }
64 
65 int main()
66 {
67     while (scanf("%d%d%d", &T, &S, &D) != EOF)
68     {
69         Init();
70         Dijkstra();
71         printf("%d\n", dist[n]);
72     }
73     return 0;
74 }