Spfa算法

Spfa算法

 

今天又写了一遍Spfa ， 感觉又熟悉了一次，怕忘了就写篇博客。

先贴篇Spfa的版子，多用就记住了。

<strong>#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<algorithm>
using namespace std;
const int Maxn = 1e5;
const int Inf = 1e8;
int vis[Maxn],dis[Maxn];
int all,N,M;
int pre[Maxn*2+5],last[Maxn*2+5],other[Maxn*2+5],cost[Maxn*2+5];
void build( int u, int v, int w ){
	pre[++all] = last[u];
	other[all] = v;
	last[u] = all;
	cost[all] = w;
}
void spfa( int x ){
	queue<int> que;
	for( int i = 1 ; i <= N ; i++ ){
		dis[i] = INF;
		vis[i] = 0;
	}
	vis[x] = 1, dis[x] = 0;
	que.push(x);
	int pos;
	while( !que.empty() ){
		pos = que.front();
		que.pop();
		int ed = last[pos];
		while( ed != -1 ){
			int dr = other[ed];
			int w = cost[ed];
			if( dis[dr] > dis[pos] + w ){
				dis[dr] = dis[pos] + w;
				if( !vis[dr] ){
					vis[dr] = 1;
					que.push(dr);
				}
			}
			ed = pre[ed];
		}
		vis[pos] = 0;
	}
}
int main()
{
	while( scanf("%d %d",&N,&M) != EOF ){
		int u,v,w;
		all = -1;
		memset( last , -1 , sizeof(last) );
		if( N == 0 && M == 0 )	break;
		for( int i = 0 ; i < M ; i++ )
		{
			scanf("%d %d %d",&u,&v,&w);
			build( u , v , w );
			build( v , u , w );
		}
		spfa(1);
		printf("%d\n",dis[N]);
	}
}</strong>

多组时dis最好memset一下，最重要的是 输入的时候 不要忘了 all = -1 ， 和memset  last = -1这两步。

计算是否是最短路的时候 ， 可以开头和结尾各 Spfa 一次，枚举之前输入的每条路径，判断它是否包含在最短路图里

下面会贴出具体代码，注意结尾枚举的时候要交换 枚举边的 两个端点 ， 才能保证 判断完全正确，这种时候的多组一定要memset dis 和 dos （这是查了好久才发现的原因）。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <iostream>
#include <cmath>
#include <map>
#include <queue>
#include <algorithm>
#include <set>
#include <vector>
using namespace std;
typedef long long LL;
const int Maxn = 1e6 + 7;
const int Inf = 1e9 + 7;
int P , T , all = -1;
int pre[Maxn*2+5] , last[Maxn*2+5] , other[Maxn*2+5] , cost[Maxn*2+5];
int dis[Maxn*2+5] , dos[Maxn*2+5] , vis[Maxn*2+5];
struct none{
	int x , y , w;
};
none Match[Maxn];

void Build( int u , int v , int w ){
	pre[++all] = last[u];
	other[all] = v;
	last[u] = all;
	cost[all] = w;
}
void INPUT (){
	all = -1;
	int u,v,w;
	memset( last , -1 , sizeof(last) );
	for( int i = 0 ; i < T ; i++ ){
		scanf(" %d %d %d",&u,&v,&w);
		Match[i].x = u , Match[i].y = v , Match[i].w = w;
		Build( u , v , w );
		Build( v , u , w );
	}
}

void Spfa ( int x ){
	queue < int > Q;
	for( int i = 0 ; i < P ; i++ ){
		if( x == 0 )
			dis[i] = Inf;
		else if( x == P-1 )
			dos[i] = Inf;
		vis[i] = 0;
	}
	vis[x] = 1;
	if( x == 0 )	dis[x] = 0;
	else	dos[x] = 0;
	int pos;
	Q.push(x);
	while( !Q.empty() ){
		pos = Q.front();
		Q.pop();
		int ed = last[pos];
		while( ed != -1 ){
			int dr = other[ed];
			int w = cost[ed];
			if( dis[dr] > dis[pos] + w || dos[dr] > dos[pos] + w ){
				if( x == 0 )	dis[dr] = dis[pos] + w;
				if( x == P - 1 )	dos[dr] = dos[pos] + w;
				if( !vis[dr] ){
					vis[dr] = 1;
					Q.push(dr);
				}
			}
			ed = pre[ed];
		}
		vis[pos] = 0;
	}
}

void Check (){
	int Sum = 0;
	for( int i = 0 ; i < T ; i++ ){
		if( dis[Match[i].x] + dos[Match[i].y] + Match[i].w == dis[P-1] || dis[Match[i].y] + dos[Match[i].x] + Match[i].w == dis[P-1] ){
			Sum += Match[i].w;
		}
	}
	cout << Sum * 2 << endl;
	memset( pre , 0 , sizeof(pre) );
	memset( other , 0 , sizeof(other) );
	memset( cost , 0 , sizeof(cost) );
	memset( vis , 0 , sizeof(vis) );
	memset( dis , 0 , sizeof(dis) );
	memset( dos , 0 , sizeof(dos) );
}

int main() {
	while( cin >> P >> T ){
		INPUT();
		Spfa(0);
		Spfa(P-1);
		Check();
	}
}

难受啊 ， debug ，反正感觉该犯得不该犯的错误基本都犯了 ， 写个博客张长记性。