Pat class B 1067 trial password (20 points)

When you try to log in to a system and forget your password , The system will generally only allow you to try a limited number of times , When the allowed number of times is exceeded , The account will be locked . This topic asks you to realize this small function .

Input format ：

Enter a password on the first line （ Length not exceeding 20 Of 、 No spaces 、Tab、 Non empty string of carriage return ） And a positive integer N（≤10）, They are the correct password and the number of attempts allowed by the system . Then each line gives a non empty string ending with carriage return , Is the password the user tried to enter . Input guarantees at least one attempt . When you read a single line # Character time , End of input , And this line is not user input .

Output format ：

Every input to the user , If the password is correct and the number of attempts does not exceed N, Output in one line Welcome in, And end the program ; If it's wrong , Then output in one line in format Wrong password: Wrong password entered by the user ; When the wrong attempt reaches N When the time , One more line Account locked, And end the program .

sample input 1：

Correct%pw 3
correct%pw
[email protected]
whatisthepassword!
Correct%pw
#

sample output 1：

Wrong password: correct%pw
Wrong password: [email protected]
Wrong password: whatisthepassword!
Account locked

sample input 2：

[email protected] 3
[email protected]
[email protected]
[email protected]
try again
#

sample output 2：

Wrong password: [email protected]
Wrong password: [email protected]
Welcome in

analysis ：

Note that the number of times the input threshold is reached is output first Wrong password Then the output Account locked.

Code ：

#include <iostream>

using namespace std;

int main() {
    string pwd, test = "";
    int times, i = 0;
    cin >> pwd >> times;

    getchar();  //  buffer 

    while (true) {
        getline(cin, test);
        if (test == "#")    //  End input 
            break;
        i++;
        if (i <= times && test == pwd) {    //  The number of tests is less than the threshold and the password is incorrect 
            cout << "Welcome in" << endl;
            break;
        } else if (i <= times && test != pwd) { //  The number of tests is less than the threshold and the password is correct 
            cout << "Wrong password: " << test << endl;
            if (i == times) {   //  The number of tests reached the threshold ,
                cout << "Account locked" << endl;
                break;
            }
        }
    }
    
    return 0;
}