牛客—最短路【算法:spfa】

链接:https://ac.nowcoder.com/acm/problem/14369
来源:牛客网

题目描述
简单暴力的题目要求:
给定一个有n个顶点(从1到n编号),m条边的有向图(其中某些边权可能为负,但保证没有负环)。请你计算从1号点到其他点的最短路。
输入描述:
第一行两个整数n, m。
接下来的m行,每行有三个整数u, v, l,表示u到v有一条长度为l的边。
输出描述:
共n-1行,第i行表示1号点到i+1号点的最短路。
示例1
输入
复制
3 3
1 2 -1
2 3 -1
3 1 2
输出
复制
-1
-2
说明
对于10%的数据,n = 2,m = 2。
对于30%的数据,n <= 5,m <= 10。
对于100%的数据,1 <= n <= 20000,1 <= m <= 200000,-10000 <= l <= 10000,保证从任意顶点都能到达其他所有顶点。

因为本题是求单元最短路,而且存在负边,因此本题所用到的算法为spfa算法
最短路问题的算法选择,可以点击这里查看

下面直接给出我的代码,代码里有分析。

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

using namespace std;

const int N = 500010, inf = 1e9;

int n, m;
int h[N], w[N], e[N], ne[N], idx;//建立邻接表
int dist[N];//表示从1号点到当前点的距离
bool st[N];//表示当前点是否在队列里,即是否被确定

void add(int a, int b, int c){
    e[idx] = b;
    w[idx] = c;
    ne[idx] = h[a];
    h[a] = idx ++;
}//确定邻接表

void spfa(){
    memset(dist, 0x3f, sizeof dist);//初始化
    dist[1] = 0;//1号点到他自己距离为0
    
    queue<int> q;//建立队列
    q.push(1);//将1放入队列里
    st[1] = true;//1号点的距离确定了
    
    while (q.size()){
        int t = q.front();//取出对头元素
        q.pop();//删除对头
        
        st[t] = false;//t元素不在队列里,标记一下
        
        for (int i = h[t]; i != -1; i = ne[i]){//遍历一遍以t为头节点的链表,更新与t相连的点的最短距离
            int j = e[i];//与i相邻的点为e[i]的值
            if (dist[j] > dist[t] + w[i]){ //松弛原理
                dist[j] = dist[t] + w[i];//更新点的距离
                if (!st[j]){//如果当前点还没有被确定,即当前点的最短距离没有被确定
                    q.push(j);//将j这个点放入队列里
                    st[j] = true;//标记一下
                }
            }
        }
    }
    
    for (int i = 2; i <= n; i ++ ) cout << dist[i] << endl;//输出答案
}
int main(){
    cin >> n >> m;
    
    memset(h, -1, sizeof h);//初始化一下表头
    
    while (m -- ){
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);//建立邻接表
    }    
    
    spfa();//调用一下函数
    
    return 0;
    
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值