HDU 2066 一个人的旅行(spfa)

Description
虽然草儿是个路痴(就是在杭电待了一年多,居然还会在校园里迷路的人,汗~),但是草儿仍然很喜欢旅行,因为在旅途中 会遇见很多人(白马王子,^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
Solution
将草儿的家看作节点0,对于草儿家相邻的城市,让其与0建距离为0的边,建完图后一遍sfpa求出0到所有点的最短距离然后找出最小值即可
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 1111
#define maxm 111111
struct edge
{
    int to;
    int cost;
    int next;
}g[maxm];
int head[maxm],tol;
int dis[maxn];
int V; 
void init() 
{
    memset(head,-1,sizeof(head));
    tol=0;
}
void add(int u,int v,int c) 
{
    g[tol].cost=c;
    g[tol].to=v;
    g[tol].next=head[u];
    head[u]=tol++;
}
void spfa(int s)
{
    bool vis[maxn];
    memset(vis,false,sizeof(vis));
    queue<int>que;
    for(int i=0;i<=V;i++)
        dis[i]=INF;
    dis[s]=0;
    vis[s]=true;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=false;
        for(int i=head[u];i!=-1;i=g[i].next)
        {
            int v=g[i].to;
            int c=g[i].cost;
            if(dis[v]>dis[u]+c)
            {
                dis[v]=dis[u]+c;
                if(!vis[v])
                {
                    vis[v]=true;
                    que.push(v);
                }
            }
        }
    }
}
int main()
{
    int T,S,D,u,v,c;
    while(~scanf("%d%d%d",&T,&S,&D))
    {
        init();
        V=0;
        while(T--)      
        {
            scanf("%d%d%d",&u,&v,&c);
            V=max(V,u);V=max(V,v);
            add(u,v,c),add(v,u,c);
        }
        while(S--)
        {
            scanf("%d",&u);
            add(0,u,0),add(u,0,0);
        }
        spfa(0);
        int ans=INF;
        while(D--)
        {
            scanf("%d",&u);
            ans=min(ans,dis[u]);
        }
        printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(HDU 2066 一个人的旅行(spfa))