C++字符输入校验问题：如何限制仅接受单个Y/N字符输入

Hey there! I get exactly what's going on here—right now, when you input something like YHTY, the cin >> answer call only grabs the first character (Y) and leaves the rest sitting in the input buffer, which is why your program still treats it as a valid "yes" input instead of triggering the error message. Let's fix this so we only accept a single Y/N (case-insensitive) and reject any input that's longer than one character or uses an invalid character.

Solution 1: Read the entire input line (simplest approach) #

The easiest way to handle this is to read the entire input line as a string instead of a single char. This lets us check both the length of the input and the character itself:

#include <iostream>
#include <string>
using namespace std;

int main() {
    string answer;
    cout << "Do you want a drink? " << endl;
    cout << "Y/N: ";
    getline(cin, answer);
    
    // First check if input is exactly one character long
    if (answer.size() == 1) {
        char response = answer[0];
        if (response == 'Y' || response == 'y') {
            cout << "Okay. I'll bring you some." << endl;
        } else if (response == 'N' || response == 'n') {
            cout << "Okay suit yourself." << endl;
        } else {
            cout << "Please type just Y/N." << endl;
        }
    } else {
        // Reject any input longer than one character
        cout << "Please type just Y/N." << endl;
    }
    return 0;
}

How this works: #

• getline(cin, answer) reads the entire line of input (including any spaces, though we don't need them here) instead of just the first character.
• We first check if the input length is exactly 1—if not, we immediately show the error message.
• If it is one character, we then check if it's a valid Y/y or N/n and respond accordingly.

Solution 2: Keep using char, but validate the input buffer #

If you prefer to stick with a char variable, you can check if there are any extra characters left in the input buffer after reading the first character. This requires clearing the buffer to avoid issues with future input:

#include <iostream>
#include <limits>
#include <cctype> // For isspace()
using namespace std;

int main() {
    char answer;
    cout << "Do you want a drink? " << endl;
    cout << "Y/N: ";
    cin >> answer;
    
    bool hasExtraInput = false;
    char nextChar;
    
    // Check for non-whitespace characters remaining in the buffer
    while (cin.peek() != '\n' && cin.peek() != EOF) {
        nextChar = cin.get();
        if (!isspace(static_cast<unsigned char>(nextChar))) {
            hasExtraInput = true;
        }
    }
    
    // Clear the rest of the input buffer up to the newline
    cin.ignore(numeric_limits<streamsize>::max(), '\n');
    
    // Only accept valid single characters with no extra input
    if (!hasExtraInput && (answer == 'Y' || answer == 'y')) {
        cout << "Okay. I'll bring you some." << endl;
    } else if (!hasExtraInput && (answer == 'N' || answer == 'n')) {
        cout << "Okay suit yourself." << endl;
    } else {
        cout << "Please type just Y/N." << endl;
    }
    return 0;
}

How this works: #

• After reading the initial char, we use cin.peek() to look at the next character in the buffer without removing it.
• We loop through any remaining characters, checking if there are any non-whitespace characters (so we ignore accidental spaces but flag things like YHTY).
• We then clear the buffer with cin.ignore() to make sure leftover input doesn't mess up any future operations.
• Finally, we only consider the input valid if there are no extra characters and it's a Y/N.

I'd recommend Solution 1 for your case—it's simpler to read and maintain, and it directly addresses the problem without having to deal with input buffer nuances.

内容的提问来源于stack exchange，提问作者Jerico