#include <iostream>
 
#include <vector>
 
using namespace std;
 
class ConcreteElementA;
class ConcreteElementB;
 
/*抽象訪問者  聲明了訪問元素對象的方法，通常為每一種類型的元素對象都提供一個訪問方法*/
class Visitor
 
{
 
public:
 
     virtual void VisitConcreteElementA(ConcreteElementA *pElementA) = 0;
     virtual void VisitConcreteElementB(ConcreteElementB *pElementB) = 0;
 
};
 
/*具體訪問者 用于定義對不同類型元素對象的操作*/
class ConcreteVisitor1 : public Visitor
 
{
public:
     void VisitConcreteElementA(ConcreteElementA *pElementA){
        // 現(xiàn)在根據(jù)傳進來的pElementA，可以對ConcreteElementA中的element進行操作
    }
 
     void VisitConcreteElementB(ConcreteElementB *pElementB){
         // 現(xiàn)在根據(jù)傳進來的pElementB，可以對ConcreteElementB中的element進行操作
    }
 
};
 
/*具體訪問者2*/
class ConcreteVisitor2 : public Visitor
{
 
public:
     void VisitConcreteElementA(ConcreteElementA *pElementA){
    }
     void VisitConcreteElementB(ConcreteElementB *pElementB){
    }
};
 
 
/*抽象元素類 聲明accept()方法，用于接受訪問者的訪問*/
class Element
 
{
public:
     virtual void Accept(Visitor *pVisitor) = 0;//accept用于接受訪問者的訪問
};
 
/*具體元素類 通過調(diào)用Visitor類的visit()方法實現(xiàn)對元素的訪問*/
class ConcreteElementA : public Element
{
public:
     void Accept(Visitor *pVisitor)//通過調(diào)用visitor對象的 visit()方法實現(xiàn)對元素對象的訪問
    {
        pVisitor->VisitConcreteElementA(this);
    }
};
 
/*具體元素類 */
class ConcreteElementB : public Element
{
public:
     void Accept(Visitor *pVisitor)
    {
     pVisitor->VisitConcreteElementB(this);
    }
};
 
// ObjectStructure類(對象結(jié)構類)，能枚舉它的元素，可以提供一個高層的接口以允許訪問者訪問它的元素
class ObjectStructure
{
public:
     void Attach(Element *pElement){
         elements.push_back(pElement);
    }
 
     void Detach(Element *pElement)   
    {
 
         vector<Element *>::iterator it = find(elements.begin(), elements.end(), pElement);
         if (it != elements.end())
         {
              elements.erase(it);
         }
    }
 
     void Accept(Visitor *pVisitor){
     // 為每一個element設置visitor，進行對應的操作
         for (vector<Element *>::const_iterator it = elements.begin(); it != elements.end(); ++it)
         {
              (*it)->Accept(pVisitor);
         }
 
}
 
int main()
{
     //實例化對象結(jié)構，用于存放元素對象，提供遍歷其內(nèi)部元素的方法
     ObjectStructure *pObject = new ObjectStructure;
      //實例化具體元素 并將創(chuàng)建好的元素放入對象結(jié)構中
     ConcreteElementA *pElementA = new ConcreteElementA;
     ConcreteElementB *pElementB = new ConcreteElementB;
     pObject->Attach(pElementA);
     pObject->Attach(pElementB);
     //實例化訪問者                                                                                                                      
     ConcreteVisitor1 *pVisitor1 = new ConcreteVisitor1;
     ConcreteVisitor2 *pVisitor2 = new ConcreteVisitor2;
     //調(diào)用accept方法 來接受訪問者對象的訪問
     pObject->Accept(pVisitor1);
     pObject->Accept(pVisitor2);
 
     if (pVisitor2) delete pVisitor2;
     if (pVisitor1) delete pVisitor1;
     if (pElementB) delete pElementB;
     if (pElementA) delete pElementA;
     if (pObject) delete pObject;
 
     return 0;
}