объекты и классы

#include <iostream>
 
class Shape
{
   public:
      virtual double square() const = 0;
      virtual ~Shape() { }   
};
 
class Trapeze : public Shape
{
   private:
      double a_;
      double b_;
      double h_;
 
   public:
      Trapeze(double a, double b, double h)
     : a_(a), b_(b), h_(h)
      {
      }
 
      virtual double square() const
      {
     return (a_ + b_) * h_ / 2;
      }
};
 
class Rectangle : public Shape
{
   private:
      double a_;
      double b_;
 
   public:
      Rectangle(double a, double b)
     : a_(a), b_(b)
      {
      }
 
      virtual double square() const
      {
     return a_ * b_;
      }
};
 
int main()
{   
   std::cout << "Input shape name('trapeze' or 'rectangle'): ";
   std::string shapeName;
   std::getline(std::cin, shapeName);
 
   Shape* shape = 0;
   if(shapeName == "trapeze")
   {
      std::cout << "Input a: ";
      double a;
      std::cin >> a;
 
      std::cout << "Input b: ";
      double b;
      std::cin >> b;
 
      std::cout << "Input h: ";
      double h;
      std::cin >> h;
      shape = new Trapeze(a, b, h);      
   }
   else if(shapeName == "rectangle")
   {
      std::cout << "Input a: ";
      double a;
      std::cin >> a;
 
      std::cout << "Input b: ";
      double b;
      std::cin >> b;
 
      shape = new Rectangle(a, b);      
   }
 
   if(shape)
      std::cout << "Square: " << shape->square() << '\n';
   else 
      std::cout << "Invalid input...\n";
   
   delete shape;
 
   return 0;
}