一个人的旅行(最短路径之SPFA算法)

一个人的旅行

点我找原题

本来以为可以用Floyd直接求出所有顶点之间的最短路径，不过数据太大会超时，依然用SPFA做了

Time Limit : 1000/1000ms (Java/Other)   Memory Limit : 32768/32768K (Java/Other)

Total Submission(s) : 45   Accepted Submission(s) : 21

Font: Times New Roman | Verdana | Georgia

Font Size: ← →

Problem 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

#include <iostream>
#include <queue>
#include <cstdio>
#define INF 1000000
#define MAXN 1005
using namespace std;
struct ArcNode
{
    int to;
    int weight;
    ArcNode *next;
};
queue<int> Q;
int n,m;
int map[MAXN][MAXN];
ArcNode *List[MAXN];
int inq[MAXN];
int dist[MAXN],path[MAXN];
void SPFA(int src)
{
    int i,u;
    ArcNode *temp;
    for(i=1;i<=n;i++)
    {
        dist[i]=INF;path[i]=src;inq[i]=0;
    }
    dist[src]=0;path[src]=src;inq[src]++;
    Q.push(src);
    while(!Q.empty())
    {
        u=Q.front();Q.pop();inq[u]--;
        temp=List[u];
        while(temp!=NULL)
        {
            int v=temp->to;
            if(dist[v]>dist[u]+temp->weight)
            {
                dist[v]=dist[u]+temp->weight;path[v]=u;
                if(!inq[v]){Q.push(v);inq[v]++;}
            }
            temp=temp->next;
        }
    }
}
int main(int argc, char *argv[])
{
    int i,j;
	int ss[MAXN],dd[MAXN];
    int u,v,w;
    int t,s,d;
    while(cin>>t>>s>>d)
    {
    	n=0;
        memset(List,0,sizeof(List));
        memset(map,0,sizeof(map));
        ArcNode *temp;
        for(i=0;i<t;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            if(u>n) n=u;
            if(v>n) n=v;
            if(map[u][v]==0||w<map[u][v])
            {
            	map[u][v]=map[v][u]=w;
            	temp=new ArcNode;
            	temp->to=v;temp->weight=w;temp->next=NULL;
            	if(List[u]==NULL) List[u]=temp;
            	else {temp->next=List[u];List[u]=temp;}
            	temp=new ArcNode;
            	temp->to=u;temp->weight=w;temp->next=NULL;
            	if(List[v]==NULL) List[v]=temp;
            	else {temp->next=List[v];List[v]=temp;}
            }
        }
        for(i=0;i<s;i++) cin>>ss[i];
        for(i=0;i<d;i++) cin>>dd[i];
        int min=INF;
        for(i=0;i<s;i++)
        {
        	SPFA(ss[i]);
        	for(j=0;j<d;j++)
        	{
	        	if(min>dist[dd[j]]) min=dist[dd[j]];
	        }
        }
        for(j=0;j<n;j++)
        {
            temp=List[j];
            while(temp!=NULL)
            {
                List[j]=temp->next;delete temp;temp=List[j];
            }
        }
        cout<<min<<endl;
    }
    return 0;
}