C++设计模式 之 “数据结构” 模式：Composite、Iterator、Chain of Resposibility

  1 #include <iostream>
  2 #include <string>
  3 #include <set>
  4 
  5 using namespace std;
  6 
  7 struct Company;
  8 
  9 using depth = unsigned;
 10 using container = set<Company*>;
 11 
 12 struct Company {
 13     virtual void display(depth d) = 0;
 14     virtual void lineDuty() = 0;
 15 };
 16 
 17 //brunch nood
 18 struct ConcreteCompany : public Company {
 19     container childrens;
 20     string _name;
 21 
 22     ConcreteCompany(string name) : _name(name) {};
 23 
 24     virtual void display(depth d) override {
 25         for (depth i = 0; i < d; i++)
 26             cout << "-";
 27         
 28         cout << _name <<endl;
 29 
 30         for(auto e : this -> childrens)
 31             e->display(d + 1);
 32     }
 33 
 34     virtual void lineDuty() override {
 35         for (auto e : childrens)
 36             e->lineDuty();
 37     }
 38 
 39     void add(Company* c) {
 40         this -> childrens.insert(c);
 41     }
 42 
 43     void remove(Company* c) {
 44         this -> childrens.erase(c);
 45     }
 46 };
 47 
 48 //leaf nood
 49 struct FinanceDepartment : public Company {
 50     string _name;
 51 
 52     FinanceDepartment(string name) : _name(name) {};
 53     
 54     virtual void display(depth d) override {
 55         for (depth i = 0; i < d; i++)
 56             cout << "-";
 57 
 58         cout << _name << endl;
 59     }
 60 
 61     virtual void lineDuty() override {
 62         cout << _name << "财务部 公司收支管理" << endl;
 63     }
 64 };
 65 
 66 //leaf nood
 67 struct HRDepartment : public Company {
 68     string _name;
 69 
 70     HRDepartment(string name) : _name(name) {};
 71 
 72     virtual void display(depth d) override {
 73         for (depth i = 0; i < d; i++)
 74             cout << "-";
 75 
 76         cout << _name << endl;
 77     }
 78 
 79     virtual void lineDuty() override {
 80         cout << _name << "人力资源部 员工招聘培训管理" << endl;
 81     }
 82 };
 83 
 84 int main() {
 85     ConcreteCompany *root = new ConcreteCompany("北京总公司");
 86     root->add(new HRDepartment("总公司人力资源部"));
 87     root->add(new FinanceDepartment("总公司财务部"));
 88 
 89     ConcreteCompany *comp = new ConcreteCompany("上海华东分公司");
 90     comp->add(new HRDepartment("华东分公司人力资源部"));
 91     comp->add(new FinanceDepartment("华东分公司财务部"));
 92     root->add(comp);
 93 
 94     ConcreteCompany *comp1 = new ConcreteCompany("南京办事处");
 95     comp1->add(new HRDepartment("南京办事处人力资源部"));
 96     comp1->add(new FinanceDepartment("南京办事处财务部"));
 97     comp->add(comp1);
 98 
 99     ConcreteCompany *comp2 = new ConcreteCompany("杭州办事处");
100     comp2->add(new HRDepartment("杭州办事处人力资源部"));
101     comp2->add(new FinanceDepartment("杭州办事处财务部"));
102     comp->add(comp2);
103 
104     cout << "结构图" << endl;
105     root->display(1);
106 
107     cout << "职责" << endl;
108     root->lineDuty();
109 
110     return 0;
111 }