написать на с++ вывести на экран фигуры заполненные звездочками. Диалог с пользователем реализовать через меню.

#include <iostream>

void printSquare(int size) { for (int i = 0; i < size; i++) { for (int j = 0; j < size; j++) { std::cout << "* "; } std::cout << std::endl; } }

void printTriangle(int size) { for (int i = 0; i < size; i++) { for (int j = 0; j <= i; j++) { std::cout << "* "; } std::cout << std::endl; } }

void printMenu() { std::cout << "1. Print square" << std::endl; std::cout << "2. Print triangle" << std::endl; std::cout << "3. Exit" << std::endl; }

int main() { int choice, size;

do {
    printMenu();
    std::cout << "Enter your choice: ";
    std::cin >> choice;

    switch (choice) {
        case 1:
            std::cout << "Enter the size of the square: ";
            std::cin >> size;
            printSquare(size);
            break;
        case 2:
            std::cout << "Enter the size of the triangle: ";
            std::cin >> size;
            printTriangle(size);
            break;
        case 3:
            std::cout << "Exiting program..." << std::endl;
            break;
        default:
            std::cout << "Invalid choice. Please try again." << std::endl;
            break;
    }

} while (choice != 3);

return 0;

}