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

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2066

题目意思：给出T条路，和草儿家相邻的城市编号，以及草儿想去的地方的编号。问从草儿家到达草儿想去的地方的最短时间是多少。

     一开始自己写的只能处理单边出发的情况，对于以下这幅图，只能处理箭头所示的方向，不能向下，于是不知道为什么出现了百年难得一遇的Runtime Error（ACCESS_VIOLATION）！   

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

 

    

 

       我的答案是INF。= =...。这个答案应该是2。之所以第一组数据能过，是因为它只需要从编号1走即能找到最小值，而上图是需要从编号2走才可以找到最小值。

       于是只能看discuss了。yk所讲的无节操！！！看完大受启发。

       dijkstra 算法本质上只能求一个点到其他所有点的最短距离。那么就需要把问题转化为单源点。可以人为地认为编号为0的点是草家，由于跟草家直接相连的点距离为0，那么可以把这些点归为源点。至于草儿想去的地方有多个，把这多个点也人为地归为终点，单独设多个点n，把这些所谓的终点连到n点里。（这个n点代码有说）。那题目就转化为从点0到点n的最短时间是多少。

    

      题目中还有一个需要注意的地方，这句话：a,b 之间可能有多条路。也就是说两点之间相连的路有多条，我们需要选择把最短的那条路的时间记录下来。

　　还有就是TLE 的情况了。if 循环里面对map[i][j]初始化时，千万不能等于maxn（1000 + 5）！TLE了n次！！！

     

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

const int INF = 1 << 20;  // INF 的设置一定要足够的大
const int maxn = 1000 + 5;
int dist[maxn], map[maxn][maxn], vis[maxn];
int T, S, D, n;

void Init()
{
    for (int i = 0; i < maxn; i++)  // 只能小于，不能等于，否则会TLE!!
    {
        for (int j = 0; j < maxn; j++)  // 只能小于，不能等于, 否则会TLE!!
            map[i][j] = (i == j ? 0 : INF);
    }
    int st, end, time;
    n = 0;
    for (int i = 0; i < T; i++)
    {
        scanf("%d%d%d", &st, &end, &time);
        if (map[st][end] > time)  // 有重边
            map[st][end] = map[end][st] = time;
        // 找出最大边的编号
        n = max(n, max(st, end));
    }
    n++;   // 假设是理想中的终点（比最大的顶点大1）的编号
    for (int i = 0; i < S; i++)
    {
        scanf("%d", &st);
        map[0][st] = map[st][0] = 0; // 草儿家到相邻城市的距离为0
    }
    for (int i = 0; i < D; i++)
    {
        scanf("%d", &st);
        map[st][n] = map[n][st] = 0;  // 想去的地方到理想中的终点距离为0
    }
}

void Dijkstra()
{
    for (int i = 0; i <= n; i++)  // 从0改为1
        dist[i] = map[0][i];  // 以草儿家和她相邻的点作为起点，求出该起点到相邻点的时间
    memset(vis, 0, sizeof(vis));
    for (int i = 0; i <= n; i++)    // 0 改为 1 也行
    {
        int u;
        int maxx = INF;
        for (int j = 0; j <= n; j++)  
        {
            if (!vis[j] && dist[j] < maxx)
                maxx = dist[u=j];
        }
        vis[u] = 1;
        for (int j = 0; j <= n; j++)
        {
            if (dist[j] > dist[u] + map[u][j])
                 dist[j] = dist[u] + map[u][j];
        }
    }
}

int main()
{
    while (scanf("%d%d%d", &T, &S, &D) != EOF)
    {
        Init();
        Dijkstra();
        printf("%d\n", dist[n]);
    }
    return 0;
}