链接: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;
}