• 组合模式为组合对象提供了与单体对象一致的处理方式，从而使得用户可以忽略两者的差异并以统一方式使用

class Shape {
 public:
  virtual void Display() = 0;
  virtual void Append(const std::shared_ptr<Shape>&) {}
  virtual void Remove(const std::shared_ptr<Shape>&) {}

  virtual std::shared_ptr<Shape> Child(std::size_t) { return nullptr; }

  virtual ~Shape() = default;
};

class Line : public Shape {  // 单体的线对象，除了 Display() 其他均使用默认实现
 public:
  void Display() override { std::cout << 1; }
};

class Rect : public Shape {  // 单体的矩形对象，除了 Display() 其他均使用默认实现
 public:
  void Display() override { std::cout << 2; }
};

class Picture : public Shape {  // 包含多个单体对象的组合对象
 public:
  void Display() override {  // 提供与单体对象一致的接口
    for (auto&& x : shapes_) {
      x->Display();
    }
  }

  void Append(const std::shared_ptr<Shape>& s) override {
    shapes_.emplace_back(s);
  }

  void Remove(const std::shared_ptr<Shape>& s) override {
    if (const auto it = std::find(std::begin(shapes_), std::end(shapes_), s);
        it != std::end(shapes_)) {
      shapes_.erase(it);
    }
  }

  std::shared_ptr<Shape> Child(std::size_t n) override { return shapes_.at(n); }

 private:
  std::vector<std::shared_ptr<Shape>> shapes_;
};

int main() {
  std::unique_ptr<Shape> picture = std::make_unique<Picture>();
  const auto line = std::make_shared<Line>();
  const auto rect = std::make_shared<Rect>();

  picture->Append(line);
  picture->Append(rect);
  picture->Display();  // 12

  picture->Remove(line);
  picture->Display();  // 2
}