单源最短路径 spfa

1、 spfa求最短路

题目描述： 给定一个n个点m条边的有向图，图中可能存在重边和自环， 边权可能为负数。
请你求出1号点到n号点的最短距离，如果无法从1号点走到n号点，则输出impossible。数据保证不存在负权回路。

题解： 正如题目描述，spfa算法适用于大部分图的单源最短路，包括存在负权边时，但是如果存在负权回路，那么求得的最短路就不一定正确了。可以说比Dijsktra用的更多，但因为 spfa 算法的复杂度不稳定 ( O(n) ~ O(n*m) ) ，所以有极个别的题目会卡掉这个算法，那时候就老老实实用堆优化Dijsktra吧（用Dijsktra时要保证没有负权边）。

模板理解： 算法核心是 BFS + queue ，但要保证queue里每个点不能同时存在多个，所以queue里存的是每个点最早的变小后的值 （有点 Dijsktra 的意思）；

Code：

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

using namespace std;

const int N = 1e5+7;

int e[N],w[N],ne[N],h[N],idx; 			//有向图
int dis[N];
bool st[N];
int n,m;

void add(int a,int b,int v){
	e[idx] = b,w[idx] = v,ne[idx] = h[a] ,h[a] = idx++;
}

int spfa()
{
	memset(dis,0x3f,sizeof dis);
	dis[1] = 0;
	queue<int> q;
	q.push(1);
	st[1] = 1;	//标记是否在队列里
	while(!q.empty()){
		int t = q.front();
		q.pop();
		st[t] = 0;		//不在队列里了
		for(int i = h[t];i!=-1;i=ne[i]){	//搜索邻边
			int j = e[i];
			if(dis[j] > dis[t] + w[i]){
				dis[j] = dis[t] + w[i];
				if(!st[j]) q.push(j) , st[j] = 1;  //每个点只在一次队列里
			}			
		}					
	}
	return dis[n];
}

int main()
{
	memset(h,-1,sizeof h);
	cin>>n>>m;
	while(m--){
		int x,y,v;
		cin>>x>>y>>v;
		add(x,y,v);
	}
	int ans = spfa();
	if(ans == 0x3f3f3f3f) cout<<"impossible\n";
	else cout<<ans<<endl;
	return 0;
}

2、spfa判断负环

题目描述： 给定一个n个点m条边的有向图，图中可能存在重边和自环， 边权可能为负数。请你判断图中是否存在负权回路。

模板理解： 见注释
复杂度： O(n*m)
Code：

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

using namespace std;

const int N = 1e5+7;

int e[N],w[N],ne[N],h[N],idx; 
int dis[N],cnt[N];
bool st[N];
int n,m;


void add(int a,int b,int v)
{
	e[idx] = b,w[idx] = v,ne[idx] = h[a] ,h[a] = idx++;
}

int spfa()
{
	queue<int> q;
	for(int i=1;i<=n;i++) q.push(i),st[i] = 1;	//由于有些点可能到不了负环 
	
	while(!q.empty()){
		int t = q.front();
		q.pop();
		st[t] = 0;
		for(int i = h[t];i!=-1;i=ne[i]){
			int j = e[i];
			if(dis[j] > dis[t] + w[i]){		// dis初始值是零，负环上点的最短路会是负无穷 
				dis[j] = dis[t] + w[i];
				cnt[j] = cnt[t] + 1;		
				if(cnt[j] >= n) return 1;	 //负环上的点更新次数 >= n ; 
				if(!st[j]) 
				q.push(j) , st[j] = 1;
			}
		}
	}
	return 0;
}

int main()
{
	memset(h,-1,sizeof h);
	cin>>n>>m;
	while(m--){
		int x,y,v;
		cin>>x>>y>>v;
		add(x,y,v);
	}
	bool ans = spfa();
	if(ans==1) puts("Yes");
	else puts("No");
	return 0;
}