G - 一个人的旅行

一个人的旅行

题目描述

虽然草儿是个路痴（就是在杭电待了一年多，居然还会在校园里迷路的人，汗~),但是草儿仍然很喜欢旅行，因为在旅途中 会遇见很多人（白马王子，^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

---

思路：

迪杰斯特算法，但是要做适当的处理：
将起点当作第一个节点，然后再假设一个终点，与题中所有 的终点相连，作为最后一个节点；

---

代码

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <string.h>

using namespace std;

#define INF 63756
const int maxn=1000+5;
int table[maxn][maxn];//邻接表
int dist[maxn],vis[maxn];//dist[i]存储从起点到i最近的路径长度

int T,S,D,n;

void init();
void dijkstra();

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

void init()
{
    int i,j;

    for(i=0; i<maxn; i++)
        for(j=0; j<maxn; j++)
        {
            table[i][j]=(i==j?0:INF);
        }

    int st,endd,time;

    n=0;

    for(i=0; i<T; i++)
    {
        scanf("%d%d%d",&st,&endd,&time);
        if(time<table[st][endd]) //避免有重复的边
            table[st][endd]=table[endd][st]=time;

        n=max(n,max(st,endd));//找到节点数
    }

    n++;//给所要到达的终点连接到一个节点上

    for(i=0; i<S; i++)
    {
        scanf("%d",&st);
        table[0][st]=table[st][0]=0;//将与起点相邻的边的权重设为0
    }

    for(i=0; i<D; i++)
    {
        scanf("%d",&endd);
        table[n][endd]=table[endd][n]=0;//将与终点相邻的边的权重设为0
    }

}

void dijkstra()
{
    int i,j;

    for(i=0; i<=n; i++)
    {
        dist[i]=table[0][i];
    }

    memset(vis,0,sizeof(vis));

    for(i=1; i<=n; i++)
    {
        int u=0;
        int minx=INF;

        for(j=0; j<=n; j++)
        {
            if(!vis[j]&&minx>dist[j])
            {
                minx=dist[j];
                u=j;
            }
        }

        vis[u]=1;

        for(j=0; j<=n; j++)
        {
            if(dist[j]>dist[u]+table[u][j])
                dist[j]=dist[u]+table[u][j];
        }
    }
}