组合模式

 1 // 组合模式.cpp : 定义控制台应用程序的入口点。
 2 //
 3 
 4 #include "stdafx.h"
 5 #include<iostream>
 6 #include<string>
 7 #include<vector>
 8 using namespace std;
 9 
10 class Component
11 {
12 public:
13     Component(){}
14     virtual ~Component(){}
15     virtual void add(Component *)=0;
16     virtual void remove(Component *)=0;
17     virtual void display(int jibie)=0;//为了看清楚树形关系而设置的显示函数
18 };
19 class Leaf:public Component
20 {
21 private:
22     string name;
23 public:
24     Leaf(string n)
25     {
26         name=n;
27     }
28     ~Leaf(){}
29     virtual void add(Component* a)
30     {
31         cout<<"最底层不能再添加部门"<<endl;
32     }
33     virtual void remove(Component* a)
34     {
35         cout<<"最底层没有可以删除的部门"<<endl;
36     }
37     virtual void display(int jibie)
38     {
39         for(int i=0;i<jibie;i++)
40         {
41             cout<<'-';
42         }
43         cout<<name<<endl;
44     }
45 };
46 class Composite:public Component
47 {
48 private:
49     vector<Component*> com;
50     string name;
51 public:
52     Composite(string n)
53     {
54         name=n;
55     }
56     virtual ~Composite(){}
57     virtual void add(Component* a)
58     {
59         com.push_back(a);
60     }
61     virtual void remove(Component*a)
62     {
63         //com.erase(a);
64     }
65     virtual void display(int num)
66     {
67         for(int i=0;i<num;i++)
68         {
69             cout<<'-';
70         }
71         cout<<name<<endl;
72         num=num+2;
73         vector<Component*>::iterator it;
74         for(it=com.begin();it!=com.end();it++)
75         {
76             (*it)->display(num);
77         }
78     }
79 };
80 
81 int _tmain(int argc, _TCHAR* argv[])
82 {
83     Component *root=new  Composite("河海大学");
84     Component *changzhouxiaoqu=new Composite("河海大学常州校区");
85     Component *wulianwang=new Composite("河海大学常州校区物联网工程学院");
86     Component *jidiangongcheng=new Composite("河海大学常州校区机电工程学院");
87     root->add(changzhouxiaoqu);
88     changzhouxiaoqu->add(wulianwang);
89     changzhouxiaoqu->add(jidiangongcheng);
90     root->display(1);
91     return 0;
92 }