PTA 1067 试密码(Python3)

当你试图登录某个系统却忘了密码时，系统一般只会允许你尝试有限多次，当超出允许次数时，账号就会被锁死。本题就请你实现这个小功能。

输入格式：

输入在第一行给出一个密码（长度不超过 20 的、不包含空格、Tab、回车的非空字符串）和一个正整数 N（≤ 10），分别是正确的密码和系统允许尝试的次数。随后每行给出一个以回车结束的非空字符串，是用户尝试输入的密码。输入保证至少有一次尝试。当读到一行只有单个 # 字符时，输入结束，并且这一行不是用户的输入。

输出格式：

对用户的每个输入，如果是正确的密码且尝试次数不超过 N，则在一行中输出 Welcome in，并结束程序；如果是错误的，则在一行中按格式输出 Wrong password: 用户输入的错误密码；当错误尝试达到 N 次时，再输出一行 Account locked，并结束程序。

输入样例 1：

Correct%pw 3
correct%pw
Correct@PW
whatisthepassword!
Correct%pw
#

输出样例 1：

Wrong password: correct%pw
Wrong password: Correct@PW
Wrong password: whatisthepassword!
Account locked

输入样例 2：

cool@gplt 3
coolman@gplt
coollady@gplt
cool@gplt
try again
#

输出样例 2：

Wrong password: coolman@gplt
Wrong password: coollady@gplt
Welcome in

提交结果：

代码：

a, b = input().split()
l = []
while True:
    data2 = input()
    if data2 == '#':
        break
    l.append(data2)
count = 1
for i in l:
    if count == int(b) and i != a:
        print('Wrong password: {}'.format(i))
        print('Account locked')
        break
    if i != a and count < int(b):
        count += 1
        print('Wrong password: {}'.format(i))
    else:
        print('Welcome in')
        break