图的接口定义

 

1. // Copyright 2008 All Rights Reserved.
2. // Author: Zhiyi ZHOU
3. #pragma once
4. #ifndef _GRAPH_H_2008_
5. #define _GRAPH_H_2008_
6. #include <assert.h>
7. #include <vector>
8. #include <queue>
9. #include <stack>
10. #include <algorithm>
11. #include <functional>
12. using namespace std;
14. class Vertex;
15. class Edge{
16. protected:
17.     int label;
18.     int weight;
19.     Vertex* source;
20.     Vertex* target;
21. public:
22.     Edge(){
23.         label=weight=0;
24.         source=target=NULL;
25.     }
26.     Edge(const Vertex& source,const Vertex& target,int label=0,int weight=0){
27.         this->label=label;
28.         this->weight=weight;
29.         this->source=const_cast<Vertex*>(&source);
30.         this->target=const_cast<Vertex*>(&target);
31.     }
32.     virtual ~Edge(){};
33.     inline virtual int getLabel()const{return label;}
34.     inline virtual void setLabel(int value){label=value;}
35.     inline virtual int getWeight()const{return weight;}
36.     inline virtual void setWeight(int value){weight=value;}
37.     inline bool operator==(const Edge& edge)const{
38.         return label==edge.getLabel();
39.     }
40.     virtual void print()const{};
41.     inline virtual Vertex& getSource()const{
42.         return *source;
43.     }
44.     virtual Vertex& setSource(Vertex& vertex){
45.         source=&vertex;
46.         return *source;
47.     }
48.     virtual Vertex& getTarget()const{
49.         return *target;
50.     }
51.     virtual Vertex& setTarget(Vertex& vertex){
52.         target=&vertex;
53.         return *target;
54.     }
55. };
56. class Vertex{
57. protected:
58.     int index;
59.     int outDegree;
60.     int inDegree;
61. public:
62.     Vertex(int index=0):index(index){
63.         outDegree=0;
64.         inDegree=0;
65.     };
66.     inline virtual int getIndex()const{return index;}
67.     inline virtual void setIndex(int index){this->index=index;}
68.     inline int getOutDegree()const{return outDegree;}
69.     inline int getInDegree()const{return inDegree;}
70.     virtual Edge* getFirstEdge()const=0;
71.     virtual Edge* getNextEdge(const Edge& edge)const=0;
72.     virtual void setInEdge(Edge* edge)=0;
73.     virtual void setOutEdge(Edge* edge)=0;
74.     virtual Edge& addEdge(Vertex& vertex,int weight,bool directed=false)=0;
75.     virtual Edge& removeEdge(const Vertex& vertex,bool directed=false)=0;
76.     virtual bool pointTo(const Vertex& vertex)const=0;
77.     virtual void print()const{}
78.     virtual ~Vertex(){}
79. };
80. class Graph
81. {
82. protected:
83.     bool directed;
84.     vector<bool> mark;
85.     vector<bool> edgeMark;
86. public:
87.     typedef unsigned int size_t;
88.     typedef Vertex& Reference;
89.     typedef const Vertex& const_Reference;
90.     Graph(bool tag=true){
91.         directed=tag;
92.     }
93.     bool isDirectedGraph()const{return directed;}
94.     
95.     virtual size_t getVerticesNum()const=0;
96.     virtual size_t getEdgeNum()const=0;
98.     virtual Reference getVertex(size_t index)const=0;
99.     virtual void setVertex(Vertex& vertex,size_t index)=0;
101.     Reference operator[](size_t index)const{
102.         return getVertex(index);
103.     }
104.     virtual ~Graph(void){}
106.     bool isVisited(const Vertex& vertex)const{
107.         return mark.at(vertex.getIndex());
108.     }
109.     void visit(const Vertex& vertex){
110.         mark[vertex.getIndex()].flip();
111.     }
112.     bool isVisited(const Edge& edge)const{
113.         return edgeMark.at(edge.getLabel());
114.     }
115.     void visit(const Edge& edge){
116.         edgeMark[edge.getLabel()]=true;
117.         if(!isDirectedGraph()){
118.             Edge* e=findEdge(edge.getTarget(),edge.getSource());
119.             if(e!=NULL)
120.                 edgeMark[e->getLabel()]=true;
121.         }
122.     }
123.     void clearState(){
124.         mark.assign(mark.size(),false);
125.         edgeMark.assign(edgeMark.size(),false);
126.     }
127. protected:
128.     Edge* findEdge(const Vertex& src,const Vertex& target)const{
129.         if(&src==NULL || &target==NULL)
130.             return NULL;
131.         size_t index=target.getIndex();
132.         for(Edge* edge=src.getFirstEdge();edge!=NULL;edge=src.getNextEdge(*edge)){
133.             if(edge->getTarget().getIndex()==index)
134.                 return edge;
135.         }
136.         return NULL;
137.     }
138.     virtual Vertex& addVertex(Vertex& vertex)=0;
139.     virtual Vertex& remove(size_t index)=0;
140. };
141. #endif