如何在c++中使用"int promptYN(string reply)“结束hangman游戏

#include <iostream>
#include <string>

using namespace std;

int promptYN(string reply) {
    cout << reply << " (Y/N): ";
    char input;
    cin >> input;

    if (input == 'Y' || input == 'y') {
        return 1; // 结束游戏
    } else if (input == 'N' || input == 'n') {
        return 0; // 继续游戏
    } else {
        cout << "Invalid input. Please enter Y or N." << endl;
        return promptYN(reply); // 重新调用函数，要求用户重新输入
    }
}

int main() {
    // 在适当的位置调用promptYN函数来判断是否结束游戏
    int endGame = promptYN("Do you want to end the hangman game?");
    
    if (endGame == 1) {
        cout << "Game ended." << endl;
        // 在这里可以添加结束游戏的逻辑
    } else {
        cout << "Game continues." << endl;
        // 在这里可以添加继续游戏的逻辑
    }
    
    return 0;
}