设计模式C++实现十五：组合模式

组合模式（Composite)：将对象组合成树的结构以表示部分-整体的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。

树可能有无数的分支，只有反复使用Composite就可以实现树状结构了。在Component中声明所有用来管理子对象的方法，其中包括Add,Remove方法，这样实现Component接口的所有子类都具备Add,Remove方法。这样做的好处就是叶结点和枝节点相对于外界没有区别，他们具备完全一致的行为接口。称为透明方式。

使用场景：当我们发现需求中是体现部分和整体的层次结构时，以及我们希望用户可以忽略组合对象与单个对象的不同，统一地使用组合结构中的所有对象时，就可以考虑使用组合模式。

下面代码使用组合模式定义了包含人力资源部和财务部这些基本对象和分公司、办事处等组合对象的类层次结构。基本对象可以被组合成更复杂的组合对象，而这个组合对象又可以被组合，这样不断的递归下去，客户端代码中，任何用到基本对象的地方都可以使用组合对象。简单的说组合模式可以让客户一致地使用组合结构和单个对象。

#ifndef COMPOSITE_H
#define COMPOSITE_H
#include<iostream>
#include<string>
#include<list>
using namespace std;
class Company
{
	//friend class ConcreteCompany;
protected:
	string name;
public:
	Company(){}
	Company(string St) :name(St){}
	virtual void Add(Company * c){};
	virtual void Remove(Company * c){};
	virtual void Display(int depth){ cout << name; };
	virtual void LineOfDuty(){};
};

class ConcreteCompany :public Company
{
	list < Company*> companys;
public:
	ConcreteCompany(string st);
	void Add(Company * c);
	void Remove(Company * c);
	void Display(int depth);
	void LineOfDuty();
};

ConcreteCompany::ConcreteCompany(string st){ name = st; companys.push_back(new Company(st)); }
void ConcreteCompany::Add(Company * c)
{
	companys.push_back(c);
}
void ConcreteCompany::Remove(Company * c)
{
	companys.remove(c);
}
void ConcreteCompany::Display(int depth)
{
	int n = depth;
	while (n)
	{
		cout << "-"; --n;
	}
	for each (Company* var in companys)
	{
		var->Display(depth + 2);
		cout << endl;
	}
}
void ConcreteCompany::LineOfDuty()
{
	for each (Company * var in companys)
	{
		var->LineOfDuty();
		cout << endl;
	}
}

class HRDepartment : public Company
{
public:
	HRDepartment(string st);
	void Add(Company * c);
	void Remove(Company * c);
	void Display(int depth);
	void LineOfDuty();
};

void HRDepartment::Add(Company * c)
{

}

HRDepartment::HRDepartment(string st)
{
	name = st;
}

void HRDepartment::Remove(Company * c)
{

}

void HRDepartment::Display(int depth)
{
	int n = depth;
	while (n)
	{
		cout << "-"; --n;
	}
	cout << name;
}
void HRDepartment::LineOfDuty()
{
	cout << name << " 员工招聘培训管理。\n";
}

class FinanceDepartment :public Company
{
public:
	FinanceDepartment(string st) { name = st; }
	void Add(Company * c){};
	void Remove(Company * c){};
	void Display(int depth);
	void LineOfDuty();
};

void FinanceDepartment::Display(int depth)
{
	int n = depth;
	while (n)
	{
		cout << "-"; --n;
	}
	cout << name;
}

void FinanceDepartment::LineOfDuty()
{
	cout << name << " 公司财务收支管理。\n";
}

#endif

#include"Composite.h"
int main()
{
	Company* root=new ConcreteCompany ("北京总公司");
	root->Add(new HRDepartment("总公司人力资源部"));
	root->Add(new FinanceDepartment("总公司财务部"));

	ConcreteCompany comp("上海华东分公司");
	comp.Add(new HRDepartment("华东分公司人力资源部"));
	comp.Add(new FinanceDepartment("华东分公司财务部"));
	root->Add(&comp);

	ConcreteCompany comp1("南京办事处");
	comp1.Add(new HRDepartment("南京办事处人力资源部"));
	comp1.Add(new FinanceDepartment("南京办事处财务部"));
	comp.Add(&comp1);

	ConcreteCompany comp2("杭州办事处");
	comp2.Add(new HRDepartment("杭州办事处人力资源部"));
	comp2.Add(new FinanceDepartment("杭州办事处财务部"));
	comp.Add(&comp2);
	root->Display(1);
	root->LineOfDuty();

	comp.Remove(&comp1);
	root->Display(1);
	root->LineOfDuty();
	return 0;
}