最短路径之SPFA算法（邻接矩阵+邻接表）

（一）SPFA算法

【前言】：上一篇的Bellman_ford算法就可以解决求图中含有负权边时的最短路径，但我们可以看到Bellman_ford算法是一个三重循环，如果数据量大的话，它的效率非常低（当然，本来效率就低）。而SPFA算法就是对Bellman_ford算法的一个优化。SPFA算法的全称是：Shortest Path Faster Algorithm。SPFA算法是西南交通大学段凡丁于1994年发表的。

【算法思想】：动态逼近法，用一个先进先出的队列来存各结点，每次取队首元素作为一个中转站，如果一个结点i直接到源点的距离大于i结点经过这个中转站再到源点的距离，就进行松弛操作，即：将dis[i]减小到dis[x]+mp[x][i]。SPFA算法也可以用来判断是否含有负权回路，如果一个结点入队次数大于节点总和数，就说明有负权回路。

1.1 邻接矩阵存图

用邻接矩阵存图唯一的好处就是简单好理解，但它太浪费空间，而且容易使算法超时；

1.1.1 核心算法

用队列来存各结点，每次取出队首元作中转站，判断这个中转站是否会缩小结点i到源点的距离,如果会，就进行松弛操作，将dis[i]减小到dis[x]+mp[x][i]，然后把没有入队元素入队，并记录它入队次数，如果一个结点的入队次数大于n次，就说明有负权回路，不能求最短路径;重复操作，直到队列为空；

queue<int> q;
	q.push(1);
	dis[1] = 0;
	vis[1] = true;
	sum[1] = 1;
	while (!q.empty())
	{
		int a = q.front();//取队列首元作中转站；
		q.pop();
		vis[a] = false;//删除标记
		for (i = 1; i <= n; i++)
		{
			if (mp[a][i] != INF)//如果这个中转站和i结点直接相连，就执行以下操作；
			{
				if (dis[i] > dis[a] + mp[a][i])//中转站会缩短i到源点的距离，dis数组就进行更新
				{
					dis[i] = dis[a] + mp[a][i];
					if (vis[i] != true)//把没有入队列的元素入队列;
					{
						q.push(i);
						vis[i] = true;
						sum[i]++;
						if (sum[i] >= n)//有负权回路；
							return true;
					}
				}
			}
	}

完整代码实现：

#include<iostream>
#include<queue>
using namespace std;
#define N 1000
#define INF 0x3f3f3f3f
int mp[N][N];//用邻接矩阵存图；
int dis[N];//记录各点到源点的距离；
bool vis[N];//标记各点是否放入队列；
int sum[N];//记录各点入队次数，如果一个点的入队次数大于节点的总个数，就说有负权回路；
void inin(int n)//初始化
{
	int i, j;
	for (i = 1; i <= n; i++)
	{
		for (j = 1; j <= n; j++)
		{
			if (i == j)
				mp[i][j] = 0;
			else
				mp[i][j] = INF;
		}
	}
}
bool SPFA(int n)
{
	int i;
	memset(vis, false, sizeof(vis));//初始化为false，表示各个节点都没有入队列；
	memset(sum, 0, sizeof(sum));//初始化，每个结点入队列次数都是0
	fill(dis, dis + N, INF);//初始化，每个结点到源点的距离的无穷大
	queue<int> q;
	q.push(1);
	dis[1] = 0;
	vis[1] = true;
	sum[1] = 1;
	while (!q.empty())
	{
		int a = q.front();//取队列首元作中转站；
		q.pop();
		vis[a] = false;//删除标记
		for (i = 1; i <= n; i++)
		{
			if (mp[a][i] != INF)//如果这个中转站和i结点直接相连，就执行以下操作；
			{
				if (dis[i] > dis[a] + mp[a][i])//中转站会缩短i到源点的距离，dis数组就进行更新
				{
					dis[i] = dis[a] + mp[a][i];
					if (vis[i] != true)//把没有入队列的元素入队列;
					{
						q.push(i);
						vis[i] = true;
						sum[i]++;
						if (sum[i] >= n)//有负权回路；
							return true;
					}
				}
			}
		}
	}
	return false;
}
int main()
{
	int n, m, w;
	int i, j, k;
	int a, b, c;
	cin >> n >> m >> w;//n个结点，m个双向正权路径；w个单向负权回路；
	inin(n);
	for (i = 0; i < m; i++)
	{
		cin >> a >> b >> c;
		mp[a][b] = mp[b][a] = c;
	}
	for (i = 0; i < w; i++)
	{
		cin >> a >> b >> c;
		mp[a][b] =-c;
		/*mp[b][a] = -c;*/
	}
	cout << "是否有负权回路：";
	if (SPFA(n))
	{	
		cout << "YES" << endl;//如果存在负权回路，就输出YES，此时无法求最短路径；
		
	}
	else
	{
		cout << "NO" << endl;
		cout << "结点n到源点的最短路径是：";
		cout << dis[n] << endl;
	}
	return 0;
}

1.2 邻接表存图

邻接表存图，就是用vector容器模拟链表，当然，也可以直接用链表，但是链表每次都需要开辟新结点，浪费时间，可能会导致算法超时；

用邻接表存图时，每次取出来的队首元素是一个结点；mp[a].push_back(node(b, c));表示a到b的距离为c；

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
#define N 1000
#define INF 0x3f3f3f3f
int dis[N];
int sum[N];
bool vis[N];
struct node
{
	int to;//该结点指向的结点；
	int v;//这条边的权值；
	node(int a, int b)//对结构体进行赋值；
	{
		to = a;
		v = b;
	}
};
vector<node> mp[N];//node类型的vector动态数组;
bool SPFA(int n)
{
	int i;
	queue<node> q;//ndoe类型的队列；
	memset(vis, false, sizeof(vis));
	memset(sum, 0, sizeof(sum));
	fill(dis, dis + N, INF);
	q.push(node(1, 0));//将源点入队
	dis[1] = 0;
	vis[1] = true;
	sum[1] = 1;
	while (!q.empty())
	{
		node t = q.front();//取出队首元；作中转站;
		q.pop();
		vis[t.to] = false;
		for (i = 0; i <mp[t.to].size(); i++)
		{
			node k = mp[t.to][i];//k.v就表示t.to到i结点的距离；
			if (k.v != INF)
			{
				if (dis[k.to] > dis[t.to] + k.v)
				{
					dis[k.to] = dis[t.to] + k.v;
					if (vis[k.to] != true)
					{
						q.push(node(k.to, k.v));
						sum[k.to]++;
						vis[k.to] = true;
						if (sum[k.to] >= n)
							return true;
					}
				}
			}
		}
	}
	return false;
}
int main()
{
	int i, j;
	int a, b, c;
	int n, m, w;
	cin >> n >> m >> w;
	for (i = 0; i < N; i++)
	{
		mp[i].clear();
	}
	for (i = 0; i < m; i++)
	{
		cin >> a >> b >> c;
		mp[a].push_back(node(b, c));
		mp[b].push_back(node(a, c));
	}
	for (i = 0; i < w; i++)
	{
		cin >> a >> b >> c;
		mp[a].push_back(node(b, -c));
	}
	cout << "是否有负权回路：";
	if (SPFA(n))
	{
		cout << "YES" << endl;
	}
	else
	{
		cout << "NO" << endl;
		cout << "结点n到源点的最短路径是：";
		cout << dis[n] << endl;
	}
	return 0;
}

测试样例：
1

输入：3个结点，2条双向正权路径，1条单向负权路径；
3 2 1
1 2 3
2 3 4
3 1 8

输出：
是否有负权回路：YES

2

输入：3个结点，3条双向正权路径，1条单向负权路径；
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3

输出：
是否有负权回路：NO
结点n到源点的最短路径是：3