c++ factorial

#include <iostream>
using namespace std;

// Function to calculate factorial
int factorial(int n) {
    if (n == 0 || n == 1) {
        return 1;
    } else {
        return n * factorial(n - 1);
    }
}

// Main function
int main() {
    // Input: the number for which factorial is to be calculated
    int num;
    cout << "Enter a number: ";
    cin >> num;

    // Output: Calculate and display the factorial
    cout << "Factorial of " << num << " = " << factorial(num);

    return 0;
}