FZU 2261 浪里个浪

 Problem 2261 浪里个浪

Accept: 293    Submit: 912
Time Limit: 1500 mSec    Memory Limit : 32768 KB

Problem Description

TonyY是一个喜欢到处浪的男人，他的梦想是带着兰兰姐姐浪遍天朝的各个角落，不过在此之前，他需要做好规划。

现在他的手上有一份天朝地图，上面有n个城市，m条交通路径，每条交通路径都是单行道。他已经预先规划好了一些点作为旅游的起点和终点，他想选择其中一个起点和一个终点，并找出从起点到终点的一条路线亲身体验浪的过程。但是他时间有限，所以想选择耗时最小的，你能告诉他最小的耗时是多少吗？

Input

 

包含多组测试数据。

输入第一行包括两个整数n和m，表示有n个地点，m条可行路径。点的编号为1 - n。

接下来m行每行包括三个整数i, j, cost，表示从地点i到地点j需要耗时cost。

接下来一行第一个数为S，表示可能的起点数，之后S个数，表示可能的起点。

接下来一行第一个数为E，表示可能的终点数，之后E个数，表示可能的终点。

0<S, E≤n≤100000，0<m≤100000，0<cost≤100。

Output

输出他需要的最短耗时。

Sample Input

4 4 1 3 1 1 4 2 2 3 3 2 4 4 2 1 2 2 3 4

Sample Output

1

Source

福州大学第十四届程序设计竞赛_重现赛

 dij会tle，spfa就Ok了QAQ

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
using namespace std;
struct node
{
    int from;
    int to;
    int w;
    int next;
} edge[100005];
int head[100005];
int dist[100005];
int vis[100005];
int n,m,cont;
void add(int from,int to,int w)
{
    edge[cont].to=to;
    edge[cont].w=w;
    edge[cont].next=head[from];
    head[from]=cont++;
}
void SPFA()
{
    for(int i=0; i<=n+1; i++)
    {
        dist[i]=0x3f3f3f3f;
    }
    dist[0]=0;
    queue<int>s;
    s.push(0);
    while(!s.empty())
    {
        int u=s.front();
        vis[u]=0;
        s.pop();
        for(int i=head[u]; i!=-1; i=edge[i].next)
        {
            int v=edge[i].to;
            int w=edge[i].w;
            if(dist[v]>dist[u]+w)
            {
                dist[v]=dist[u]+w;
                if(vis[v]==0)
                {
                    s.push(v);
                    vis[v]=1;
                }
            }
        }
    }
}
void init()
{
    cont=0;
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        init();
        for(int i=0; i<m; i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
        }
        int p,q,k;
        scanf("%d",&p);
        while(p--)
        {
            scanf("%d",&k);
            add(0,k,0);
        }
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d",&k);
            add(k,n+1,0);
        }
        SPFA();
        printf("%d\n",dist[n+1]);
    }
    return 0;
}