最短路径·三:SPFA算法

描述

万圣节的晚上,小Hi和小Ho在吃过晚饭之后,来到了一个巨大的鬼屋!

鬼屋中一共有N个地点,分别编号为1..N,这N个地点之间互相有一些道路连通,两个地点之间可能有多条道路连通,但是并不存在一条两端都是同一个地点的道路。

不过这个鬼屋虽然很大,但是其中的道路并不算多,所以小Hi还是希望能够知道从入口到出口的最短距离是多少?

输入

每个测试点(输入文件)有且仅有一组测试数据。

在一组测试数据中:

第1行为4个整数N、M、S、T,分别表示鬼屋中地点的个数和道路的条数,入口(也是一个地点)的编号,出口(同样也是一个地点)的编号。

接下来的M行,每行描述一条道路:其中的第i行为三个整数u_i, v_i, length_i,表明在编号为u_i的地点和编号为v_i的地点之间有一条长度为length_i的道路。

对于100%的数据,满足N<=10^5,M<=10^6, 1 <= length_i <= 10^3, 1 <= S, T <= N, 且S不等于T。

对于100%的数据,满足小Hi和小Ho总是有办法从入口通过地图上标注出来的道路到达出口。

输出

对于每组测试数据,输出一个整数Ans,表示那么小Hi和小Ho为了走出鬼屋至少要走的路程。

Sample Input

5 10 3 5
1 2 997
2 3 505
3 4 118
4 5 54
3 5 480
3 4 796
5 2 794
2 5 146
5 4 604
2 5 63

Sample Output

172

运用vector 邻接表(spfa):

#include <iostream>
#include <queue>
#include <cstring>
using namespace std;

struct node
{
    int to,cost;
    node (int _to=0,int _cost=0) {to=_to;cost=_cost;}
};

#define inf 1e9

vector<node>vec[100005];
int dis[100005];

int n,m;

int spfa(int s)
{
    int ok=1;
    int vis[100005];
    int cis[100005];
    queue<int>que;
    memset(vis,0,sizeof(vis));
    memset(cis,0,sizeof(cis));


    for(int i=1; i<=n; i++)
    {
        dis[i]=inf;
    }

    dis[s]=0;
    que.push(s);
    vis[s]=1;
    cis[s]++;

    while (!que.empty())
    {
        int k=que.front();
        que.pop(),vis[k]=0;

        for(int i=0; i<(int)vec[k].size(); i++)
        {
            int v=vec[k][i].to;
            if(dis[v]>dis[k]+vec[k][i].cost)
            {

                dis[v]=dis[k]+vec[k][i].cost;

                if(vis[v]==0)
                {
                    que.push(v);
                    vis[v]=1;
                    cis[v]++;

                    if(cis[v]>n)
                       return 0;
                }

            }

        }

    }

    return ok;
}

int main()
{
    int s,t;

    while (cin>>n>>m)
    {

    cin >>s>>t;
    for(int i=1; i<=n; i++)
        vec[i].clear();

    for(int i=1; i<=m; i++)
    {
        int a,b,c;

        cin >>a>>b>>c;

            vec[a].push_back(node(b,c));
            vec[b].push_back(node(a,c));

    }

    if(spfa(s)==1)
        cout <<dis[t]<<endl;
    else
        cout <<"存在负权回路"<<endl;

    }


    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41199502/article/details/81869287