1 #include <iostream>
  2 #include <string>
  3 #include <vector>
  4 using namespace std;
  5 // 抽象的部件类描述将来所有部件共有的行为
  6 class Component
  7 {
  8 public:
  9      Component(string name) : m_strCompname(name){}
 10      virtual ~Component(){}
 11      virtual void Operation() = 0;
 12      virtual void Add(Component *) = 0;
 13      virtual void Remove(Component *) = 0;
 14      virtual Component *GetChild(int) = 0;
 15      virtual string GetName()
 16      {
 17           return m_strCompname;
 18      }
 19      virtual void Print() = 0;
 20 protected:
 21      string m_strCompname;
 22 };
 23 class Leaf : public Component
 24 {
 25 public:
 26      Leaf(string name) : Component(name)
 27      {}
 28      void Operation()
 29      {
 30           cout<<"I'm "<<m_strCompname<<endl;
 31      }
 32      void Add(Component *pComponent){}
 33      void Remove(Component *pComponent){}
 34      Component *GetChild(int index)
 35      {
 36           return NULL;
 37      }
 38      void Print(){}
 39 };
 40 class Composite : public Component
 41 {
 42 public:
 43      Composite(string name) : Component(name)
 44      {}
 45      ~Composite()
 46      {
 47           vector<Component *>::iterator it = m_vecComp.begin();
 48           while (it != m_vecComp.end())
 49           {
 50                if (*it != NULL)
 51                {
 52                     cout<<"----delete "<<(*it)->GetName()<<"----"<<endl;
 53                     delete *it;
 54                     *it = NULL;
 55                }
 56                m_vecComp.erase(it);
 57                it = m_vecComp.begin();
 58           }
 59      }
 60      void Operation()
 61      {
 62           cout<<"I'm "<<m_strCompname<<endl;
 63      }
 64      void Add(Component *pComponent)
 65      {
 66           m_vecComp.push_back(pComponent);
 67      }
 68      void Remove(Component *pComponent)
 69      {
 70           for (vector<Component *>::iterator it = m_vecComp.begin(); it != m_vecComp.end(); ++it)
 71           {
 72                if ((*it)->GetName() == pComponent->GetName())
 73                {
 74                     if (*it != NULL)
 75                     {
 76                          delete *it;
 77                          *it = NULL;
 78                     }
 79                     m_vecComp.erase(it);
 80                     break;
 81                }
 82           }
 83      }
 84      Component *GetChild(int index)
 85      {
 86           if (index > m_vecComp.size())
 87           {
 88                return NULL;
 89           }
 90           return m_vecComp[index - 1];
 91      }
 92      void Print()
 93      {
 94           for (vector<Component *>::iterator it = m_vecComp.begin(); it != m_vecComp.end(); ++it)
 95           {
 96                cout<<(*it)->GetName()<<endl;
 97           }
 98      }
 99 private:
100      vector<Component *> m_vecComp;
101 };
102 int main(int argc, char *argv[])
103 {
104      Component *pNode = new Composite("Beijing Head Office");
105      Component *pNodeHr = new Leaf("Beijing Human Resources Department");
106      Component *pSubNodeSh = new Composite("Shanghai Branch");
107      Component *pSubNodeCd = new Composite("Chengdu Branch");
108      Component *pSubNodeBt = new Composite("Baotou Branch");
109      pNode->Add(pNodeHr);
110      pNode->Add(pSubNodeSh);
111      pNode->Add(pSubNodeCd);
112      pNode->Add(pSubNodeBt);
113      pNode->Print();
114      Component *pSubNodeShHr = new Leaf("Shanghai Human Resources Department");
115      Component *pSubNodeShCg = new Leaf("Shanghai Purchasing Department");
116      Component *pSubNodeShXs = new Leaf("Shanghai Sales department");
117      Component *pSubNodeShZb = new Leaf("Shanghai Quality supervision Department");
118      pSubNodeSh->Add(pSubNodeShHr);
119      pSubNodeSh->Add(pSubNodeShCg);
120      pSubNodeSh->Add(pSubNodeShXs);
121      pSubNodeSh->Add(pSubNodeShZb);
122      pNode->Print();
123      // 公司不景气，需要关闭上海质量监督部门
124      pSubNodeSh->Remove(pSubNodeShZb);
125      if (pNode != NULL)
126      {
127           delete pNode;
128           pNode = NULL;
129      }
130      return 0;
131 }