组合模式是结构型设计模式的一种，核心目标是将对象组合成树形结构以表示部分-整体的层次结构，同时让客户端对单个对象和组合对象的使用具有一致性，这在处理树形结构时非常实用，尤其是在需要统一叶子节点和容器节点操作接口的场景中。
组合模式的核心设计思路
组合模式的关键在于抽象出一个公共接口，这个接口需要同时覆盖叶子节点和容器节点的核心操作。叶子节点是没有子节点的单个对象，容器节点则可以包含多个子节点，这些子节点既可以是叶子节点也可以是其他容器节点。通过让两种节点都实现同一个接口，客户端就不需要判断当前操作的是叶子还是容器，直接使用统一的接口方法即可。
C++中组合模式的实现步骤
1. 定义抽象组件接口
抽象组件是所有叶子节点和容器节点的公共基类，需要声明所有公共操作的方法。如果部分方法只有容器节点需要实现，叶子节点可以给出默认空实现或者抛异常，具体根据业务需求决定。
#include <iostream>
#include <vector>
#include <string>
#include <memory>
// 抽象组件接口
class Component {
public:
virtual ~Component() = default;
// 公共操作：显示节点信息
virtual void display(int depth) = 0;
// 容器节点的操作：添加子节点，叶子节点默认空实现
virtual void add(std::shared_ptr<Component> component) {
std::cout << "当前节点不支持添加子节点" << std::endl;
}
// 容器节点的操作：移除子节点，叶子节点默认空实现
virtual void remove(std::shared_ptr<Component> component) {
std::cout << "当前节点不支持移除子节点" << std::endl;
}
};
2. 实现叶子节点类
叶子节点继承抽象组件接口，只需要实现自身支持的操作即可，不需要支持添加、移除子节点的方法，直接使用基类的默认实现就可以。
// 叶子节点类
class Leaf : public Component {
private:
std::string name;
public:
explicit Leaf(const std::string& n) : name(n) {}
void display(int depth) override {
// 根据深度输出缩进，展示叶子节点名称
for (int i = 0; i < depth; ++i) {
std::cout << "-";
}
std::cout << "叶子节点: " << name << std::endl;
}
};
3. 实现容器节点类
容器节点需要维护一个子节点集合，实现添加、移除子节点的方法，同时在显示操作时递归调用所有子节点的显示方法，从而输出整个子树的结构。
// 容器节点类
class Composite : public Component {
private:
std::string name;
std::vector<std::shared_ptr<Component>> children;
public:
explicit Composite(const std::string& n) : name(n) {}
void display(int depth) override {
// 输出当前容器节点信息
for (int i = 0; i < depth; ++i) {
std::cout << "-";
}
std::cout << "容器节点: " << name << std::endl;
// 递归调用所有子节点的显示方法
for (auto& child : children) {
child->display(depth + 2);
}
}
void add(std::shared_ptr<Component> component) override {
children.push_back(component);
}
void remove(std::shared_ptr<Component> component) override {
for (auto it = children.begin(); it != children.end(); ++it) {
if (*it == component) {
children.erase(it);
break;
}
}
}
};
4. 客户端使用示例
客户端只需要面向抽象组件接口编程，不需要区分叶子节点和容器节点的类型，就可以完成整个树形结构的操作。
int main() {
// 创建根容器节点
auto root = std::make_shared<Composite>("根节点");
// 创建叶子节点添加到根节点
auto leaf1 = std::make_shared<Leaf>("叶子1");
auto leaf2 = std::make_shared<Leaf>("叶子2");
root->add(leaf1);
root->add(leaf2);
// 创建子容器节点
auto subComposite = std::make_shared<Composite>("子容器");
auto leaf3 = std::make_shared<Leaf>("叶子3");
subComposite->add(leaf3);
// 将子容器节点添加到根节点
root->add(subComposite);
// 统一调用display方法，无需区分节点类型
std::cout << "树形结构展示:" << std::endl;
root->display(0);
// 尝试给叶子节点添加子节点，会触发默认提示
leaf1->add(std::make_shared<Leaf>("无效叶子"));
return 0;
}
组合模式的优势与适用场景
使用组合模式统一叶子与容器的接口后，客户端代码会变得更加简洁，不需要大量的类型判断逻辑。这种模式非常适合处理文件系统、菜单结构、组织架构等天然的树形结构场景。不过需要注意，如果叶子节点和容器节点的差异非常大，强行统一接口可能会导致叶子节点实现很多无意义的空方法，此时需要结合业务场景判断是否适用。
注意事项
- 抽象组件接口的方法设计要兼顾叶子节点和容器节点的共性，避免接口过于臃肿
- 如果树形结构需要频繁遍历或者修改，建议使用智能指针管理子节点生命周期，避免内存泄漏
- 递归调用显示或者操作时，要注意树形结构的深度，避免过深导致栈溢出