软件设计⑩|组合模式

一、效果如下：

二、类图如下：

三、代码如下：

#include <iostream> 
#include <string> 
#include <list> 
using namespace std;

class AbstractFile
{
protected:
    string name;
public:
    AbstractFile(string name) :name(name) {}
    virtual void AddFile(AbstractFile* file) {}
    virtual void RemoveFile(AbstractFile* file) {}
    virtual void GetChild(int depth) {}
};

class File : public AbstractFile
{
public:
    File(string name) :AbstractFile(name) {}
    void AddFile(AbstractFile* file)
    {
        cout << "该文件不支持添加节点操作" << endl;
    }
    void RemoveFile(AbstractFile* file)
    {
        cout << "该文件不支持移除节点操作" << endl;
    }
    void GetChild(int depth)
    {
        string _tmpstring(depth, '-');
        cout << _tmpstring << name << endl;
    }
};

class Folder :public AbstractFile
{
private:
    list<AbstractFile*> _list;
public:
    Folder(string name) :AbstractFile(name) {}
    void AddFile(AbstractFile* file)
    {
        _list.push_back(file);
    }
    void RemoveFile(AbstractFile* file)
    {
        _list.remove(file);
    }
    void GetChild(int depth)
    {
        string tmpstring(depth, '-');
        cout << tmpstring << name << endl;
        list<AbstractFile*>::iterator iter = _list.begin();
        for (; iter != _list.end(); iter++)
        {
            (*iter)->GetChild(depth + 2);
        }
    }
};

class Image :public AbstractFile
{
private:
    list<AbstractFile*> _list;
public:
    Image(string name) :AbstractFile(name) {}
    void GetChild(int depth)
    {
        string _tmpstring(depth, '-');
        cout << _tmpstring << name << endl;
    }
};

class Txt :public AbstractFile
{
private:
    list<AbstractFile*> _list;
public:
    Txt(string name) :AbstractFile(name) {}
    void GetChild(int depth)
    {
        string _tmpstring(depth, '-');
        cout << _tmpstring << name << endl;
    }
};

int main()
{
    Folder* root = new Folder("root");
    Folder* user = new Folder("user");
    root->AddFile(user);
    Folder* user1 = new Folder("user1");
    Image* image1 = new Image("image1");
    Txt* txt2 = new Txt("txt2");
    user->AddFile(user1);
    user1->AddFile(image1);
    user1->AddFile(txt2);
    Folder* user2 = new Folder("user2");
    Image* image2 = new Image("image2");
    user->AddFile(user2);
    user2->AddFile(image2);
    Folder* bin = new Folder("bin");
    Txt* txt1 = new Txt("txt1");
    user->AddFile(bin);
    bin->AddFile(txt1);


    root->GetChild(1);
    cout << endl;

    delete root;
    delete user;
    delete user1;
    delete user2;
    delete bin;
    delete image1;
    delete image2;
    delete txt1;
    delete txt2;

    system("pause");
    return 0;
}

 

参考链接：https://www.jb51.net/article/80789.htm