两种语言实现设计模式（C++和Java）（十二：组合模式）

组合模式，将对象组合成属性结构以表示‘部分-整体’的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。其UML图如下：

主要解决：它在我们树型结构的问题中，模糊了简单元素和复杂元素的概念，客户程序可以像处理简单元素一样处理复杂元素，从而使得客户程序与复杂元素的内部结构解耦。
如何解决：树枝和树叶实现统一接口，树枝内部组合该接口。
关键代码：树枝内部组合该接口，并且含有内部属性list，里面放Component。

可以用组合模式来表述大学里学院之间的关系，例如工程学院包含机械学院和计算机学院，代码如下：

C++代码：

#include <iostream>
#include <vector>
#include <list>

using namespace std;

class College{
protected:
    string name;
public:
    College(string _name){
        name = _name;
    }
    virtual void add(College* p) = 0;
    virtual void remove(string name) = 0;
    virtual void display() = 0;
    string getName(){
        return this->name;
    }
};

class ConcreteCollege: public College{
public:
    ConcreteCollege(string _name): College(_name){}
    void add(College* p){
        shared_ptr<College> temp(p);
        listCollege.push_back(temp);
    }
    void remove(string name){
        auto it = listCollege.begin();
        for(it; it != listCollege.end(); it++){
            shared_ptr<College> temp(*it);
            string strName = temp->getName();
            if(name == strName){
                listCollege.erase(it);
            }
        }
    }
    void display(){
        cout << this->getName()<< endl;
        for (auto it = listCollege.begin(); it != listCollege.end(); it++){
            (*it)->display();
        }
    }
private:
    vector<shared_ptr<College>> listCollege;
};

int main()
{
    College *c1 = new ConcreteCollege("Engineering College");
    c1->add(new ConcreteCollege("Mechanical College"));
    c1->add(new ConcreteCollege("Computer College"));
    c1->display();
    return 0;
}

这里使用了shared_ptr智能指针避免delete带来的麻烦

 

Java代码

public interface ICollege {
    public void add(ICollege iCollege);
    public void delete(ICollege iCollege);
    public void display();
}

public class College implements ICollege {

    private String name;

    private ArrayList<ICollege> listCollege;

    public String getName(){return this.name;}

    public College(String name){
        listCollege = new ArrayList<ICollege>();
        this.name = name;
    }

    public void add(ICollege iCollege) {
        listCollege.add(iCollege);
    }

    public void delete(ICollege iCollege) {
        listCollege.remove(iCollege);
    }

    public void display() {
        System.out.println(name);
        for (ICollege iCollege:
             listCollege) {
            iCollege.display();
        }
    }
}

public class Main {
    public static void main(String[] args) {
        ICollege c = new College("Engineering College");
        c.add(new College("Mechanical College"));
        c.add(new College("Computer College"));
        c.display();
    }
}