How to Take Operator as Input in C++?

// C++ program to take operator as input
#include <iostream>
using namespace std;
int main()
{
    // Declaring variables to store two numbers and the
    // operator
    double num1, num2;
    char op;
    // Get the first number from the user
    cout << "Enter the first number: ";
    cin >> num1;
    // Get the second number from the user
    cout << "Enter the second number: ";
    cin >> num2;
    // Get the arithmetic operator from the user
    cout << "Enter the arithmetic operator (+, -, *, /): ";
    cin >> op;
    // Declare a variable to store the result of the
    // operation
    double result;
    // Use a switch statement to perform the operation based
    // on the operator
    switch (op) {
    case '+':
        result = num1 + num2;
        break;
    case '-':
        result = num1 - num2;
        break;
    case '*':
        result = num1 * num2;
        break;
    case '/':
        // Check if the second number is not zero before
        // performing division
        if (num2 != 0) {
            result = num1 / num2;
        }
        else {
            cout << "Error: Division by zero." << endl;
            return 1; // Return an error code
        }
        break;
    default:
        cout << "Error: Invalid operator." << endl;
        return 1; // Return an error code
    }
    // Display the result of the operation
    cout << "Result: " << result << endl;
    return 0;
}