设计模式之composite模式

组合模式又称：对象树（Object Tree）是一种结构型设计模式，可以使用它将对象组合成树状结构，并且能像使用独立对象一样使用它们。

• 组件（Component）：定义组合模式中所有对象共有的方法和属性。
• 叶子节点（Leaf）：叶子节点对象，也是组合中没有子节点的对象。
• 组合节点（Composite）：表示组合中的容器对象，该对象包含其他子节点。

实现

#include <iostream>
#include <list>
#include <memory>
#include <utility>
#include <cstddef>
 
using namespace std;
 
class Company
{
public:
	Company(string name) { m_name = name; }
	virtual ~Company() {}
	virtual void Add(std::unique_ptr<Company>) {}//这里默认实现为空函数，leaf节点中不用实现为空了
	virtual void Show(int depth) {} // 这里默认实现为空函数，leaf节点中不用实现为空了
protected:
	string m_name;
};
 
//具体公司
class ConcreteCompany : public Company
{
public:
	ConcreteCompany(string name) : Company(name) {}
	virtual ~ConcreteCompany() 
    {
		for (auto& company : m_listCompany)
		{
		     company.reset(nullptr);
		}
	}
	void Add(std::unique_ptr<Company> pCom) { m_listCompany.push_back(std::move(pCom)); } //位于树的中间，可以增加子树
	void Show(int depth)
	{
		for (int i = 0; i < depth; i++)
			cout << "-";
		cout << m_name << endl;
		auto iter = m_listCompany.begin();
		for (; iter != m_listCompany.end(); iter++) //显示下层结点
			(*iter)->Show(depth + 2);
	}
private:
	list<std::unique_ptr<Company>> m_listCompany;
};
 
//具体的部门，财务部
class FinanceDepartment : public Company
{
public:
	FinanceDepartment(string name) :Company(name) {}
	virtual ~FinanceDepartment() {}
	virtual void Show(int depth) //只需显示，无限添加函数，因为已是叶结点
	{
		for (int i = 0; i < depth; i++)
			cout << "-";
		cout << m_name << endl;
	}
};
 
//具体的部门，人力资源部
class HRDepartment :public Company
{
public:
	HRDepartment(string name) :Company(name) {}
	virtual ~HRDepartment() {}
	virtual void Show(int depth) //只需显示，无限添加函数，因为已是叶结点
	{
		for (int i = 0; i < depth; i++)
			cout << "-";
		cout << m_name << endl;
	}
};
 
int main()
{
	auto root = std::make_unique<ConcreteCompany>("总公司");
	auto leaf1 = std::make_unique < FinanceDepartment>("总公司财务部");
	auto leaf2 = std::make_unique < HRDepartment>("总公司人力资源部");
	root->Add(std::move(leaf1));
	root->Add(std::move(leaf2));
 
	//分公司
	auto mid1 = std::make_unique < ConcreteCompany>("杭州分公司");
	auto leaf3 = std::make_unique < FinanceDepartment>("杭州分公司财务部");
	auto leaf4 = std::make_unique < HRDepartment>("杭州分公司人力资源部");
	mid1->Add(std::move(leaf3));
	mid1->Add(std::move(leaf4));
	root->Add(std::move(mid1));
	//分公司
	auto mid2 = std::make_unique < ConcreteCompany>("上海分公司");
	auto leaf5 = std::make_unique < FinanceDepartment>("上海分公司财务部");
	auto leaf6 = std::make_unique < HRDepartment>("上海分公司人力资源部");
	mid2->Add(std::move(leaf5));
	mid2->Add(std::move(leaf6));
	root->Add(std::move(mid2));
	root->Show(0);
 
	
	return 0;
}