问题描述
给定一个n个顶点,m条边的有向图(其中某些边权可能为负,但保证没有负环)。请你计算从1号点到其他点的最短路(顶点从1到n编号)
输入格式
第一行两个整数n, m。
接下来的m行,每行有三个整数u, v, l,表示u到v有一条长度为l的边。
输出格式
共n-1行,第i行表示1号点到i+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算法解决
SPFA算法可以解决带负权的单源最短路径问题,并且能够判断是否存在负权环
Dijkstra算法只能解决不带负权的单源最短路径问题
#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;
const int MAXD = 0xFFFFFFF;
const int MAXN = 20000+5;
struct edge{
int start;
int end;
int value;
edge(int u,int v,int l){
this->start = u;
this->end = v;
this->value = l;
}
};
vector<edge> e[MAXN];
int dis[MAXN];
int vis[MAXN]; //用于记录该结点当前是否在队中。0为不在。元素一出队vis立刻置为0,根据vis判断下一个结点是否应入队。vis的这种策略可以反映元素的情况,无需自定义队列的查找函数
int num[MAXN]; //用于记录该点入队了几次,如果大于n说明存在负权环
queue<int> q;
int n,m,u,v,l;
bool spfa(){
bool res = true;
dis[1] = 0;
q.push(1);
num[1]++;
vis[1] = 1;
while(!q.empty()){
int head = q.front();
q.pop();
vis[head] = 0;
vector<edge> now = e[head];
for(int i = 0;i<now.size();i++){
if(dis[head] + now[i].value < dis[now[i].end]){
dis[now[i].end] = dis[head] + now[i].value;
if(vis[now[i].end] == 0){
q.push(now[i].end);
num[now[i].end]++;
if(num[now[i].end] > n)
res = false;
}
}
}
}
return res;
}
int main(){
cin >> n >> m;
for(int i = 0;i<m;i++){
cin >> u >> v >> l;
e[u].push_back(edge(u,v,l));
}
for(int i = 1;i<=n;i++)
dis[i] = MAXD;
memset(num,0,sizeof(num));
memset(vis,0,sizeof(vis));
if(spfa()){
for(int i = 2;i<=n;i++){
if(i != 2)
cout << endl;
cout << dis[i];
}
}
return 0;
}