拓扑排序的方法有两种(是我知道的有两种):一种是利用深度优先搜索,把节点状态分成未发现(WHITE),发现(GRAY)和其相邻节点被访问完(BLACK)。根据BLACK时间按从大到小的顺序排序,即可以得到拓扑排序。第二种方法是把所有节点的入度做一个小顶堆,然后不断删除一条边,如果度为0了,则出堆,最后能够构成拓扑排序的结果。
代码如下:
//GraphMatrix.h
#ifndef __GraphMatrix__
#define __GraphMatrix__
#include<iostream>
#include<vector>
#include<list>
#include<stack>
using namespace std;
struct Vertex
{
enum COL{WHITE,GRAY,BLACK};
char data;
int discover;
int finish;
COL color;
Vertex(char d=0)
{
data=d;
discover=0;
finish=0;
color=WHITE;
}
};
struct Edge
{
int weight;
Edge(int w=INT_MAX)
{
weight=w;
}
};
class GraphMatrix
{
int n;
int e;
vector<Vertex> V;//顶点集合
vector<vector<Edge> >E;//边的集合
bool TSort(Vertex& v,int& time,stack<Vertex>&);
public:
GraphMatrix()
{
n=e=0;
}
int insertVertex(Vertex & v);
void insertEdge(Vertex from,Vertex to,Edge e);
void removeEdge(Vertex from,Vertex to);
void removeVertex(Vertex &v);
stack<Vertex> ToPoLogicalSort(Vertex v);
};
#endif
//GraphMatrix.cpp
#include"GraphMatrix.h"
#include<queue>
#include<stack>
using namespace std;
bool GraphMatrix::TSort(Vertex& v,int &time,stack<Vertex> &result)
{
if(v.color!=Vertex::WHITE)
return true;
time+=1;
int k=-1;
for(int i=0;i<V.size();i++)
if(V[i].data==v.data)
{
k=i;
break;
}
if(k==-1)
return false;
if(V[k].color==Vertex::WHITE)
{
V[k].discover=time;
V[k].color=Vertex::GRAY;
cout<<v.data<<endl;
}
for(int i=0;i<E[k].size();i++)
{
if(E[k][i].weight!=INT_MAX && V[i].color==Vertex::GRAY)
return false;
if(E[k][i].weight!=INT_MAX && V[i].color==Vertex::WHITE)
{
if(TSort(V[i],time,result)==false)
return false;
}
}
V[k].color=Vertex::BLACK;
time+=1;
V[k].finish=time;
result.push(V[k]);
}
int GraphMatrix::insertVertex(Vertex & v)
{
V.push_back(v);
E.push_back(vector<Edge>(V.size()));
if(E.size()>1)
for(int i=0;i<V.size()-1;i++)
E[i].push_back(Edge());
n++;
return V.size()-1;
}
void GraphMatrix::insertEdge(Vertex from,Vertex to,Edge e)
{
int k=0;
int p=-1,q=-1;
for(int i=0;i<V.size();i++)
{
if(V[i].data==from.data)
{
k+=1;
p=i;
}
if(V[i].data==to.data)
{
k+=2;
q=i;
}
}
if((k&1)==0)
p=insertVertex(from);
if((k&2)==0)
q=insertVertex(to);
E[p][q]=e;
// E[q][p]=e;//如果不屏蔽则是无向图
this->e++;
}
void GraphMatrix::removeEdge(Vertex from,Vertex to)
{
int k=0;
int p=-1,q=-1;
for(int i=0;i<V.size();i++)
{
if(V[i].data==from.data)
p=i;
if(V[i].data==to.data)
q=i;
}
if(p==-1 || q==-1)
return ;
else
E[p][q]=INT_MAX;
n--;
}
void GraphMatrix::removeVertex(Vertex &v)
{
int k=-1;
for(int i=0;i<V.size();i++)
{
if(V[i].data==v.data)
k=i;
}
if(k==-1)
return ;
V.erase(V.begin()+k);
for(int i=0;i<E.size();i++)
E[i].erase(E[i].begin()+k);
E.erase(E.begin()+k);
n--;
}
stack<Vertex> GraphMatrix::ToPoLogicalSort(Vertex v)
{
for(int i=0;i<V.size();i++)
V[i].color=Vertex::WHITE;
int time=0;
stack<Vertex> result;
if(TSort(v,time,result)==false)
return stack<Vertex>();
for(int i=0;i<V.size();i++)
if(V[i].color==Vertex::WHITE)
{
if(TSort(V[i],time,result)==false)
return stack<Vertex>();
}
return result;
}
#include<iostream>
#include<stack>
#include"GraphMatrix.h"
using namespace std;
#define DEBUG
int main()
{
#ifdef DEBUG
freopen("input.txt","r",stdin);
#endif
GraphMatrix gm;
char vertex,from,to;
int e;
//测试边和顶点的插入
cout<<"输入边"<<endl;
while(cin>>from>>to>>e)
{
cout<<"输入边"<<endl;
gm.insertEdge(from,to,e);//调用默认构造函数
}
//深度优先搜索
stack<Vertex> result=gm.ToPoLogicalSort('A');
cout<<endl;
while(!result.empty())
{
cout<<result.top().data<<" ";
result.pop();
}
system("pause");
return 0;
}
拓扑排序的函数时TopoLogicalSort()。