Problem 2261 浪里个浪几许情愁

Problem 2261 浪里个浪

Accept: 261 Submit: 818
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

思路：找一个超级起点（就是某点（如0点）到所有起点的距离），找一个超级终点（就是所有终点到某点（如n+1）的距离），添加结构体里，再来一边spaf

具体操作看代码：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<math.h>
#include<map>
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
int m,n,e;
int dis[155555],mark[155555],f[155555];

struct p
{
    int a,b,n;
}q[555555];

void add(int a,int b,int c)
{
    q[e].a=b;
    q[e].b=c;
    q[e].n=f[a];
    f[a]=e++;
}

void spaf()
{
    int u,v;
    queue<int>que;
    que.push(0);
    for(int i=0;i<=n+1;i++)//这里注意一下！用for循环初始dis数组，用memset函数会wa，我也搞不懂
        dis[i]=inf;//以后大家都用fou循环为好
    memset(mark,0,sizeof(mark));
    dis[0]=0;
    while(!que.empty())
    {
        u=que.front();
        mark[u]=0;
        que.pop();
        for(int i=f[u];i!=-1;i=q[i].n)
        {
            v=q[i].a;
            if(dis[v]>dis[u]+q[i].b)
            {
                dis[v]=dis[u]+q[i].b;
                if(mark[v]==0)
                {
                    mark[v]=1;
                    que.push(v);
                }
            }
        }
    }
    printf("%d\n",dis[n+1]);
}
int main()
{
    while(~scanf("%d %d",&n,&m))
    {
        int i,j,S,E,u,v,w;
        memset(f,-1,sizeof(f));
        e=0;
        for(i=1;i<=m;i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            add(u,v,w);
        }
        scanf("%d",&S);//将0点设置为超级起点，连接0点到所有起点
        while(S--)
        {
            scanf("%d",&v);
            add(0,v,0);
        }
        scanf("%d",&E);//将n+1点设置为超级终点，链接所有终点到n+1点
        while(E--)
        {
            scanf("%d",&u);
            add(u,n+1,0);
        }
        spaf();//然后就是spaf
    }
}