设计模式 - 组合模式

设计模式- 组合模式

组合模式

定义：将对象组合成树形结构以表示“部分和整体”的层次结构，组合模式使得用户对单个对象和组合对象的使用具有一致性.

优点：高层模块调用简单，组合模式使得客户端代码可以一致地处理单个对象和组合对象，无须关心自己处理的是单个对象还是组合对象。
缺点：在使用组合模式时，其子节点和树枝节点都是实现类，而不是接口，违反了依赖倒置原则，设计起来麻烦。

实现

#include<cstdio>
#include<cmath>
#include<map>
#include<vector>
#include<stack>
#include<iostream>
#include<algorithm>
#include<string>
const int maxn=1e5+10;
const int INF=0x3f3f3f3f;
using namespace std;
class Component{
	public:
		Component(string name) {a_name = name;}
		virtual ~Component(){}
		virtual void Add(Component* c){}
		virtual void Display(int depth){}
	protected:
		string a_name;	
}; 
class Leaf:public Component{
	public:
		Leaf(string name):Component(name){}
		virtual void Add(Component* c){
			cout<<"Cannot add a leaf"<<endl;
		}
		void Display(int depth){
			for(int i=0;i<depth;i++) cout<<"-";
			cout<<" "<<a_name<<endl;
		}
		virtual ~Leaf(){}
};
class Reaf:public Component{
	public:
		Reaf(string name):Component(name){}
		virtual ~Reaf(){}
		void Add(Component* c){
			cout<<"Cannot add a Reaf"<<endl;
		}
		void Display(int depth){
			for(int i=0;i<depth;i++) cout<<"-";
			cout<<" "<<a_name<<endl;
		}
}; 
class Composite : public Component{
	private:
		vector<Component *>children;
		vector<Component *>::iterator iter;
	public:
		Composite(string name): Component(name){}
		virtual ~Composite(){}
		virtual void Add(Component* c){
			children.push_back(c);
		}
		void Display(int depth){
			for(int i=0;i<depth;i++) cout<<"-";
			cout<<" "<<a_name<<endl;
			for(iter=children.begin();iter!=children.end();iter++){
				(*iter)->Display(depth+2);
			}
			
		}
};
int main(){
	Component *root = new Composite("company");
	Component *leaf1 = new Leaf("leaf1");
	Component *leaf2 = new Leaf("leaf2");
	root->Add(leaf1);
	root->Add(leaf2);
	Component *mid = new Composite("midCompany");
	Component *leaf3 = new Leaf("leaf3");
	Component *leaf4 = new Leaf("leaf4");
	mid->Add(leaf3);
	mid->Add(leaf4);
	root->Add(mid);
	Component *mid2 = new Composite("mid2Company");
	Component *leaf5 = new Leaf("leaf5");
	Component *leaf6 = new Leaf("leaf6");
	mid2->Add(leaf5);
	mid2->Add(leaf6);
	root->Add(mid2);
	Component *mid3 = new Composite("mid3Company");
	Component *leaf7 = new Leaf("leaf7");
	Component *leaf8 = new Leaf("leaf8");
	mid3->Add(leaf7);
	mid3->Add(leaf8);
	mid2->Add(mid3);
	root->Display(0);
}