设计模式复习-组合模式

#pragma once
#include "stdafx.h"
#include<set>
#include<string>
#include<iostream>
using namespace std;

/*
设计模式-组合模式（Composite）[用类建一棵树]
    将对象组合成树形结构以表示 部分-整体 的层次结构。组合模式使得用户对
单个对象和组合对象的使用具有一致性。
*/

class CComponent {//所有节点统一接口
protected:
	string m_strName;
public:
	CComponent(const string &strName) {
		m_strName = strName;
	}
	virtual void Add(CComponent * const pc) = 0;
	virtual void Remove(CComponent * const pc) = 0;
	virtual void Display(const int & nDepth) = 0;
};

class CLeaf : public CComponent {
public:
	CLeaf(const string &strName) : CComponent(strName) {}
	void Add(CComponent *const pc) {
		cout << "Cannot add to a leaf" << endl;
	}
	void Remove(CComponent *const pc) {
		cout << "Cannot remove from a leaf" << endl;
	}
	void Display(const int &nDepth) {
		cout << "-" << nDepth << ":" << m_strName << endl;
	}
};

class CCpmposite : public CComponent {
private:
	set<CComponent*>m_cChildren;
public:
	CCpmposite(const string &strName) : CComponent(strName) {
		m_cChildren.clear();
	}
	void Add(CComponent * const pc) {
		m_cChildren.insert(pc);
	}
	void Remove(CComponent * const pc) {
		m_cChildren.erase(pc);
		delete pc;
	}
	void Display(const int &nDepth) {
		cout << "-" << nDepth << ":" << m_strName << endl;
		for each(auto i in m_cChildren) {
			i->Display(nDepth + 1);
		}
	}
	~CCpmposite() {
		for each(auto i in m_cChildren) {
			delete i;
		}
	}
};


int main() {

	CCpmposite *pRoot = new CCpmposite("root");
	pRoot->Add(new CLeaf("Lead A"));
	pRoot->Add(new CLeaf("Lead B"));

	CCpmposite *pComp = new CCpmposite("Composite X");
	pComp->Add(new CLeaf("Lead A"));
	pComp->Add(new CLeaf("Lead B"));

	pRoot->Add(pComp);

	pRoot->Display(1);

	delete pRoot;

	getchar();
	return 0;
}