SPFA, Dijkstra 代码实现

这里给出了 Bellmanford 和 SPFA 的算法思想和伪代码, 但这离具体的实现还有差距, 最短路径算法的使用频率很高, 有必要总结一下

 

1. SPFA 算法

算法的核心是: 松弛有效的操作必然发生在松弛前导节点成功松弛的节点上

/*
 * spfa.h
 *
 *  Created on: Apr 8, 2014
 *      Author: sangs
 */

#ifndef SPFA_H_
#define SPFA_H_
#include <queue>
using namespace std;

extern class Edge;
const int MAXNUM = 1000;
int dist[MAXNUM];
int father[MAXNUM];
int mark[MAXNUM];
vector<Edge> graph[MAXNUM];

void spfa(int st, int ed) {
	queue<int> record;
	record.push(st);
	dist[st] = 0;
	father[st] = -1;

	while(!record.empty()) {
		int u = record.front().index;
		record.pop();

		for(int i = 0; i < graph[u].size(); i ++) {
			int v = graph[u][i].index;
			if(dist[u] + graph[u][i].cost < dist[v]) {
				record.push(v);
				father[v] = u;
				dist[v] = dist[u] + graph[u][i].cost;
			}
		}
	}

	for(int i = ed; i != -1; i = father[i]) {
		mark[i] = 1;
	}
}


#endif /* SPFA_H_ */

　　

2. Dijkstra 算法

代码的核心是对 visited 赋值在外面, 这样能够保证 st->ed 的路径是最短的

/*
 * Dijkstra.h
 *
 *  Created on: Apr 8, 2014
 *      Author: sangs
 */

#ifndef DIJKSTRA_H_
#define DIJKSTRA_H_

extern class Node;
const int MAXNUM = 1000;
bool visited[MAXNUM];
/*
 * Always used to calculate the shortest path/length to specific node
 */
void dijkstra(int st, int ed) {
	priority_queue<int> record;

	record.push(st);

	while(!record.empty()) {
		int u = record.top();
		
		while(!record.empty()) {
			Node v = record.top();
			record.pop();
			visited[v.index] = true;
			if(v == ed) {
				break;
			}
			
			for(int i = 0; i < graph[v.index].size(); i ++) {
				if(visited(...)) continue;
				
				record.push_back(Node(...));
			}
		}
	}
}


#endif /* DIJKSTRA_H_ */

　　

 

2014年4月18日21:51:42

上面的 SPFA 代码有误, 需要判断一个点是否两次被加入队列