[C++]组合模式

组合模式（Composite Pattern），又叫部分整体模式，是用于把一组相似的对象当作一个单一的对象。组合模式依据树形结构来组合对象，用来表示部分以及整体层次。这种类型的设计模式属于结构型模式，它创建了对象组的树形结构。

github源码路径：https://github.com/dangwei-90/Design-Mode

 // 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

// 参考大话设计模式 - 组合模式

#include <iostream>
#include <vector>

#ifndef SAFE_DELETE
#define SAFE_DELETE(p) { if(p){delete(p); (p)=NULL;} }
#endif

using namespace std;

class Component {
public:
	Component() {}
	Component(string name) {
		name_ = name;
	}

	virtual void Add(Component* component) = 0;
	virtual void Remove(Component* component) = 0;
	virtual void Display(int depth) = 0;

public:
	string name_ = "";
};

// 叶子结点
class Leaf : public Component {
public:
	Leaf(string name) {
		name_ = name;
	}

	void Add(Component* component) {
		cout << "not have a component" << endl;
	}

	void Remove(Component* component) {
		cout << "not have a component" << endl;
	}

	void Display(int depth) {
		for (int n = 0; n < depth; n++) {
			cout << "-";
		}
		cout << name_ << endl;
	}; 
};

// 结点
class Composite : public Component {
public:
	Composite(string name) {
		name_ = name;
	}

	void Add(Component* component) {
		component_vector_.push_back(component);
	}

	void Remove(Component* component) {
		component_vector_.emplace_back(component);
	}

	// 遍历该结点，以及该结点拥有的子节点
	void Display(int depth) {
		for (int n = 0; n < depth; n++) {
			cout << "-";
		}
		cout << name_ << endl;

		for (auto iter = component_vector_.begin(); iter != component_vector_.end(); iter++) {
			(*iter)->Display(depth + 2);
		}
	}

private:
	vector<Component*> component_vector_;
};

int main()
{
	// 根节点及两个叶子结点
	Composite* root = new Composite("root");
	root->Add(new Leaf("leaf_a"));
	root->Add(new Leaf("leaf_b"));

	// 子结点一，及两个叶子结点
	Composite* comp_1 = new  Composite("comp_1");
	comp_1->Add(new Leaf("comp1_a"));
	comp_1->Add(new Leaf("comp1_b"));

	// 子结点一挂在根节点上
	root->Add(comp_1);
	
	// 子结点二，及两个叶子结点
	Composite* comp_2 = new  Composite("comp_2");
	comp_2->Add(new Leaf("comp2_a"));
	comp_2->Add(new Leaf("comp2_b"));

	// 子结点二挂在子结点一上
	comp_1->Add(comp_2);

	// 子结点三，及两个叶子结点
	Composite* comp_3 = new  Composite("comp_3");
	comp_3->Add(new Leaf("comp3_a"));
	comp_3->Add(new Leaf("comp3_b"));

	// 子结点三挂在根节点上
	root->Add(comp_3);

	// 从根节点开始遍历
	root->Display(1);

	SAFE_DELETE(comp_3);
	SAFE_DELETE(comp_2);
	SAFE_DELETE(comp_1);
	SAFE_DELETE(root);

	return 0;
}