约束输入长度python sys.stdin.readline（）

Question

我尝试解决this website中的算法问题！问题是所有问题都必须由系统输入来解决。

所以我会用

from stdin import sys
a = stdin.readline()
# do something

我的问题是有时候我需要像下面这样。 第一行是整数n（1n个具有n位数字的字符串或数字的长度。

让我给你举个例子。

输入

4
1100

问题是如何约束匹配4位数字（长度）的第二行作为输入。怎么样？以及为什么？

Answer 1

您不能以此方式限制输入长度。

按照@Andrey的建议，您应该阅读输入，如果不正确，请打印错误并阅读新的输入。

reading input in python 3的Python方式是：

a = input("prompt")

因此您可以执行以下操作：

a = ""
while len(a) != 4:
    a = input("please enter a 4 digit number: ")