Bellman-ford算法能解决有负权边的问题,完美的解决了迪杰特斯拉算法的缺点。核心代码:
for (int i=1; i<=n-1; i++)
{
bool flag = false;
for (int j=1; j<=m; j++)
{
if (dis[v[j]] > dis[u[j]] + cost[j])
{
dis[v[j]] = dis[u[j]] + cost[j];
flag = true;
}
}
if (!flag)
{
break;
}
}
它只循环n-1次,每次都对所有边进行松弛操作,正因为如此,对于n个顶点的图,只需要循环n-1,那么如果到第n次的时候还能够松弛的话,那这张图肯定有负权回路。所以上面代码可以把循环改为n,在i == n时判断是否还能松弛。需要注意的问题是v[i]一定是弧尾也就是有向边箭头指向的点。具体为什么画图想一下就能明白了。
下面的代码是三个数组,可以写为一个结构体。
完整代码如下:
#include <iostream>
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAX_N = 50001;
int main()
{
int u[MAX_N], v[MAX_N], w[MAX_N], dis[MAX_N];
int n, m;
scanf("%d %d", &n, &m);
for (int i=0; i<m; i++)
{
scanf("%d %d %d", &u[i], &v[i], &w[i]);
}
for (int i=1; i<=n; i++)
{
dis[i] = INF;
}
dis[1] = 0;
for (int k=1; k<=n-1; k++)
{
int ck = 0;
for (int i=0; i<m; i++)
{
if (dis[v[i]] > dis[u[i]] + w[i])
{
dis[v[i]] = dis[u[i]]+w[i];
ck = 1;
}
}
if (ck == 0)
{
break;
}
}
int flag = 0;
for (int i=1; i<=m; i++)
{
if (dis[v[i]] > dis[u[i]] + w[i])
{
flag = 1;
}
}
if (flag == 1)
{
cout << "有负边!" << endl;
}
else
{
for (int i=1; i<=n; i++)
{
cout << dis[i] << " " << endl;
}
}
return 0;
}