適合使用並行的一種bfs

這種寫法的bfs和前面的最大區別在於它對隊列的處理,以前的簡單bfs是每次從隊列中取出當前的訪問節點後,以後就將它的鄰接節點加入到隊列中,這樣明顯不利於並行化,node

爲此,這裏使用了兩個隊列,第一個隊列上當前同一層的節點,第二個隊列用來存儲當前層節點的全部鄰接節點,等到當前層的節點所有訪問完畢後,再將第一個隊列與第二個隊列進行交換,便可。ios

這樣作的優點在於便於之後的並行化。同一層的節點能夠一塊兒運行,不會受到下一層節點的干擾。spa

#include <stdio.h>
#include <queue>
#include <map>
#include <iostream>
#include <stdlib.h>
#include <omp.h>
#include <string>
#include <getopt.h>
#include "CycleTimer.h"
#include "graph.h"
#include "bfs.h"
using namespace std;
void clear(queue<int>& q) {
    queue<int> empty;
    swap(empty, q);
}
int main(){
	string graph_filename="../../Data/facebook.txt";
	graph g;
	load_graph(graph_filename.c_str(),&g);
	printf("Graph stats:\n");
	printf("  Edges: %d\n", g.num_edges);
	printf("  Nodes: %d\n", g.num_nodes);
	queue<int> q1,q2;
	int * distance=(int *)malloc(sizeof(int)*g.num_nodes);
	for(int i=0;i<g.num_nodes;i++){
		distance[i]=-1;
	}
	q1.push(1);
	distance[1]=0;
	while(!q1.empty()){
		clear(q2);
		for(int i=0;i<q1.size();i++){
			int node=q1.front();
			q1.pop();
			cout<<node<<"->";
			int start_edge=g.outgoing_starts[node];
			int end_edge=(node==g.num_nodes-1)?g.num_edges:g.outgoing_starts[node+1];
			for(int j=start_edge;j<end_edge;j++){
				int outgoing=g.outgoing_edges[j];
				if(distance[outgoing]==-1){
					distance[outgoing]=1;
					q2.push(outgoing);
				}
			}
		}
		swap(q1, q2);
	}
	return 1;
}
相關文章
相關標籤/搜索