c++实现组合模式（附带源码）

最新推荐文章于 2025-06-05 17:52:55 发布

Katie。

最新推荐文章于 2025-06-05 17:52:55 发布

阅读量786

点赞数 8

分类专栏： c、c++实战项目文章标签： c++ 组合模式开发语言

本文链接：https://blog.csdn.net/m0_61840987/article/details/145727389

版权

c、c++实战项目专栏收录该内容

516 篇文章

订阅专栏

C++ 实现组合模式（Composite Pattern）

项目介绍

组合模式（Composite Pattern）是一种结构型设计模式，旨在将对象组合成树形结构以表示“部分-整体”层次。它允许客户端以统一的方式处理单个对象和对象集合，从而简化了代码的复杂性。组合模式在处理树形结构数据时非常有用，能够通过统一的接口来操作树节点和树本身。

什么是组合模式

组合模式的核心思想是通过递归的方式将对象组合成一个树形结构，使得客户端能够一致地对待单个对象和对象集合。组合模式的目的是使得客户端无需关注对象是单一的还是复合的，它们的处理方式是一样的。

在组合模式中，我们通常会有两类对象：

叶子节点（Leaf）：表示树的最基本元素，它不能包含其他对象。
组合节点（Composite）：表示树的中间节点，包含叶子节点或其他组合节点。

组合模式的关键在于，组合节点和叶子节点都实现了相同的接口，从而客户端可以统一地对待它们。

项目背景

假设我们正在设计一个文件管理系统，该系统包含文件和文件夹。文件是最基本的元素，而文件夹可以包含文件或其他文件夹。通过组合模式，我们可以方便地处理文件和文件夹的层级关系，统一对待它们的操作（如计算总大小、显示信息等）。

项目实现思路

定义组件接口：首先定义一个 Component 类，作为叶子节点和组合节点的公共接口。
实现叶子节点类：叶子节点代表文件，它只需要实现 Component 接口的基本方法。
实现组合节点类：组合节点代表文件夹，它也实现 Component 接口，并持有子组件（文件或文件夹）的集合。
提供统一的操作接口：通过组合模式的接口，客户端可以像操作单个对象一样操作整个树形结构。

项目实现代码

#include <iostream>
#include <vector>
#include <memory>
#include <string>

// 组件接口
class Component {
public:
    virtual void showDetails() const = 0;  // 显示详细信息
    virtual int getSize() const = 0;       // 获取大小
    virtual ~Component() = default;        // 虚析构函数
};

// 叶子节点：文件
class File : public Component {
private:
    std::string name;
    int size;

public:
    File(const std::string& name, int size) : name(name), size(size) {}

    void showDetails() const override {
        std::cout << "文件名: " << name << ", 大小: " << size << "KB" << std::endl;
    }

    int getSize() const override {
        return size;
    }
};

// 组合节点：文件夹
class Folder : public Component {
private:
    std::string name;
    std::vector<std::shared_ptr<Component>> children;

public:
    Folder(const std::string& name) : name(name) {}

    void add(const std::shared_ptr<Component>& component) {
        children.push_back(component);
    }

    void showDetails() const override {
        std::cout << "文件夹名: " << name << std::endl;
        for (const auto& child : children) {
            child->showDetails();  // 递归显示文件夹内容
        }
    }

    int getSize() const override {
        int totalSize = 0;
        for (const auto& child : children) {
            totalSize += child->getSize();  // 递归计算总大小
        }
        return totalSize;
    }
};

// 客户端代码
int main() {
    // 创建文件对象
    std::shared_ptr<Component> file1 = std::make_shared<File>("file1.txt", 10);
    std::shared_ptr<Component> file2 = std::make_shared<File>("file2.txt", 20);
    std::shared_ptr<Component> file3 = std::make_shared<File>("file3.txt", 30);

    // 创建文件夹对象
    std::shared_ptr<Component> folder1 = std::make_shared<Folder>("folder1");
    folder1->add(file1);
    folder1->add(file2);

    std::shared_ptr<Component> folder2 = std::make_shared<Folder>("folder2");
    folder2->add(file3);
    folder2->add(folder1);  // 将folder1添加到folder2中

    // 显示文件夹及其内容
    std::cout << "显示 folder2 的详细信息:" << std::endl;
    folder2->showDetails();

    // 计算 folder2 的总大小
    std::cout << "\nfolder2 的总大小: " << folder2->getSize() << "KB" << std::endl;

    return 0;
}