設計模式二三事——組合模式

三、組合模式

組合模式允許派生類包含基類實例，以此來構成相互嵌套的組合方式，可構成類似樹形結構或者線性表，具體代碼如下：

//the base composite
class Company
{
public:
    Company( char* name )
    {
        m_p = p;
    }

    virtual void Add( Company* pCom ){}
    virtual void Show( int depth ){}

protected:
    char* m_p;
};

class FinanceDepartment : public Company
{
public:
    FinanceDepartment( char* name ) : Company( name )
    {

    }

    virtual void Show( int depth )
    {
        for( int i = 0; i < depth; ++i )
        {
            cout<<"-";
        }
        cout<<m_p<<endl;
    }
};

class HRDepartment : public Company
{
public:
    HRDepartment( char* name ) : Company( name )
    {

    }

    virtual void Show( int depth )
    {
        for( int i = 0; i < depth; ++i )
        {
            cout<<"-";
        }
        cout<<m_p<<endl;
    }
};

class ConcreteCompany : public Company
{
public:
    ConcreteCompany( char* name ) : Company( name )
    {
        m_p = name;
    }

    virtual void Add( Company* pCom )
    {
        m_list.push_back( pCom );
    }

    virtual void Show( int depth )
    {
        for( int i = 0; i < depth; ++i )
        {
            cout<<"-";
        }
        cout<<m_p<<endl;
        for( vector< Company* >::size_type i = 0; i < m_list.size(); ++i )
        {
            m_list[ i ]->Show( depth + 2 );
        }
    }

private:
    vector< Company* > m_list;               //the container of the composite
};

上述代碼中，公司類是一個基本抽象類，而金融部門和HR部門是派生類，具體的公司可以包含金融部門和HR部門，而HR部門裏面有可以包含各自的金融部門或者金融部門可以包含各自的HR部門，實現了組合的效果。