编程语言
首页 > 编程语言> > 算法基础之搜索与图论——SPFA算法( 队列优化过的bellman-ford算法 ) ( 单源最短路,存在负权边,适用于各种图 ) 时间复杂度O(m),最坏O(nm)

算法基础之搜索与图论——SPFA算法( 队列优化过的bellman-ford算法 ) ( 单源最短路,存在负权边,适用于各种图 ) 时间复杂度O(m),最坏O(nm)

作者:互联网

题目:SPFA求最短路

给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环, 边权可能为负数。

请你求出 1 号点到 n 号点的最短距离,如果无法从 1 号点走到 n 号点,则输出 impossible。

数据保证不存在负权回路。

输入格式
第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z。

输出格式
输出一个整数,表示 1 号点到 n 号点的最短距离。

如果路径不存在,则输出 impossible。

数据范围
1≤n,m≤10^5,
图中涉及边长绝对值均不超过 10000。

输入样例:
3 3
1 2 5
2 3 -3
1 3 4

输出样例:
2

思想:bellman-ford算法无脑遍历所有边,直到没有边的dist改变为止,而根据dist[j] = dist[t] + w得知 如果dist[i]没有改变那么dist[j] 就不会发生改变,所以我们可以将改变过的点压入队列,然后对队列中的点进行遍历,如果发现遍历之后改变了其他点,那么就将其他点也压入队列,如此往复直到队列为空。

代码如下:

#include<iostream>
#include<queue>
#include<cstring>

using namespace std;

const int N = 100005;

int n, m;
//以下数组和之前代码作用一样
int h[N],e[N], ne[N], w[N], flag;
int dist[N];
//st为标志数组,标志点是否存在队列中
bool st[N];
//建图
void add(int x, int y, int z)
{
    e[flag] = y;
    w[flag] = z;
    ne[flag] = h[x];
    h[x] = flag;
    flag ++;
}

void spfa()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;
    
    queue<int> q;
    q.push(1);
    
    st[1] = true;
    
    while(q.size()){
        int t = q.front();
        q.pop();
        //st变为false表示点t已不在队列
        st[t] = false;
        //对t的邻接点进行遍历,
        for(int i = h[t]; i != -1; i = ne[i]){
            int j = e[i];
            
            if(dist[j] > dist[t] + w[i]){
                dist[j] = dist[t] + w[i];
                //如果此点被更新了且不在队列之中,那么就将此点压入队列
                if(!st[j]){
                    q.push(j);
                    st[j] = true;
                }
            }
        }
    }
    //因为这里不是遍历所有边,而是遍历改变过的点的邻接边
    //所以dist[n]如果不存在是不可能被遍历到的,因此判断条件不用为dist[n] > 0x3f3f3f3f / 2
    if(dist[n] == 0x3f3f3f3f)   cout << "impossible";
    else    cout << dist[n];
}
int main()
{
    cin >> n >> m;
    
    memset(h, -1, sizeof h);
    while(m --){
        int x, y, z;
        cin >> x >> y >> z;
        
        add(x, y, z);
    }
    
    spfa();
}

标签:遍历,dist,int,单源,st,队列,算法,负权,号点
来源: https://blog.csdn.net/qq_48813213/article/details/115443327