這種寫法的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; }