以下代码来源: 设计模式精解-GoF 23种设计模式解析附C++实现源码
//Component.h
#pragma once class Component { public: Component(); virtual ~Component(); virtual void Operation() = 0; virtual void Add(const Component&); virtual void Remove(const Component&); virtual Component* getChild(int); protected: private: };
//Component.cpp
#include"Component.h" Component::Component(){} Component::~Component(){} void Component::Add(const Component& com){} void Component::Remove(const Component& com){} Component* Component::getChild(int index) { return 0; }
//composite.h
#include"Component.h" #include<vector> class Composite :public Component { public: Composite(); virtual ~Composite(); void Add(Component* com); void Remove(Component* com); void Operation(); Component* Getchild(int index); protected: private: std::vector<Component*>comVec; };
//Composite.cpp
#include"Component.h" #include"composite.h" const int null = 0; Composite::Composite(){} Composite::~Composite(){} void Composite::Operation(){ for (std::vector<Component*>::iterator comIter = comVec.begin(); comIter != comVec.end(); ++comIter) { (*comIter)->Operation(); } } void Composite::Add(Component* com) { comVec.push_back(com); } void Composite::Remove(Component* com) { //comVec.erase(&com);//此处有问题,求解释!!! } Component* Composite::Getchild(int index) { return comVec[index]; }
//Leaf.h
#include"Component.h" class Leaf :public Component { public: Leaf(); virtual ~Leaf(); void Operation(); protected: private: };
//Leaf.cpp
#include"Leaf.h" #include<iostream> Leaf::Leaf(){ } Leaf::~Leaf(){} void Leaf::Operation(){ std::cout << "Leaf Operation..." << std::endl; }
//main.cpp
#include"Component.h" #include"composite.h" #include"Leaf.h" #include<iostream> #include<string> int main(int args, char* argv) { Leaf* I = new Leaf(); I->Operation(); Composite* com = new Composite(); com->Add(I); com->Operation(); Component* II = com->Getchild(0); getchar(); return 0; }