FZU 2261 浪里个浪

公开20180602远古博客，汗

题目链接：FZU 2261

 

Problem 2261 浪里个浪

Accept: 287    Submit: 903
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 41 3 11 4 22 3 32 4 42 1 22 3 4

 Sample Output

1

 

 

解题历程：

        （n——点数，m——边数）        

        由于对dijkstra比较熟悉，一开始是想用dijkstra做的，直接遍历每个起点会T掉，设了个超级源点0，无奈还是T掉了，（只想着中间会跳出，太菜，都没想着仔细推一推时间复杂度，O(n^2)完美T掉了），之后问了下队友试没试过dijkstra，各种轰炸，想起来之前听说过的堆优化可以降到mlogn（但看起来比spfa难写，难理解，于是先投奔spfa的怀抱，关于堆优化是否可行，自己还没试过）。

        spfa的km复杂度看起来很优秀呐，但关于k的值也是众说纷纭，spfa不稳定，点反复入队出队，会被卡数据。

        （PS：因为INF赋值wa了几发，INF=（n-1）*边长上限，因为要保证所有可达路径长< INF，刚开始竟然忘了*（n-1））

 

解题思路：

        设了个超级源点0，为了方便再来个超级汇点n+1，0到可以做起点的点cost=0，可以做终点的点到n+1的cost=0。之后跑spfa，听有经验的学长说邻接表比vector快，就把自己之前的vector换成邻接表了（被T怕了。。。正好邻接表打得也不是很熟，练练手）。

 

        感觉代码写得罗嗦了（看大神们代码，改良代码风格去了，以后会一点一点变完善的）。

 

AC代码：

#include<cstdio>
#include<vector>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int INF=1e7+5;
int first[300005];
int next [300005];
int w[300005],u[300005],v[300005];
int dist[100005];
int vis[100005];
queue<int>q;
int n;

void init()
{
    for (int i=0; i<=n+1; i++)
    {
        first[i]=-1;
        vis[i]=0;
        next[i]=-1;
        dist[i]=INF;
        w[i]=INF;
    }
}

int spfa()
{
    while (!q.empty())
    {
        q.pop();
    }
    q.push(0);
    vis[0]=1;
    dist[0]=0;
    while (!q.empty())
    {
        int now =q.front();
        q.pop();
        vis[now]=0;
        for (int id=first[now];id!=-1;id=next[id])
        {
            if (dist[v[id]]>dist[now]+w[id])
            {
                dist[v[id]]=dist[now]+w[id];
                if (!vis[v[id]])
                {
                    q.push(v[id]);
                    vis[v[id]]=1;
                }
            }
        }
    }
    return dist[n+1];
}

int main()
{
    int m;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        for (int i=1; i<=m; i++)
        {
            scanf("%d%d%d",&u[i],&v[i],&w[i]);
            next[i]=first[u[i]];
            first[u[i]]=i;
        }
        int frcnt,tocnt;
        scanf("%d",&frcnt);
        for (int i=1; i<=frcnt; i++)
        {
            scanf("%d",&v[m+i]);
            u[m+i]=0;
            next[m+i]=first[0];
            first[0]=m+i;
            w[m+i]=0;
        }
        scanf("%d",&tocnt);
        for (int i=1; i<=tocnt; i++)
        {
            scanf("%d",&u[m+frcnt+i]);
            v[m+frcnt+i]=n+1;
            next[m+frcnt+i]=first[u[m+frcnt+i]];
            first[u[m+frcnt+i]]=m+frcnt+i;
            w[m+frcnt+i]=0;
        }
        printf("%d\n",spfa());
    }
}