Определив, является ли данное число нечетным или четным, создайте функцию, которая возвращает квадрат, если он четный, или куб, если нет, в качестве результата.C++

#include <iostream>

#include <cmath>

int squareOrCube(int num) {

   if(num % 2 == 0) {

       return pow(num, 2); // возврат квадрата числа

   } else {

       return pow(num, 3); // возврат куба числа

   }

}

int main() {

   int number;

   std::cout << "Введите число: ";

   std::cin >> number;

   int result = squareOrCube(number);

   std::cout << "Результат: " << result << std::endl;

   return 0;

}