約束輸入長度python sys.stdin.readline（）

Question

我試圖從這個網站解決算法問題！ 問題是所有問題都必須通過系統輸入來解決。

所以我會用

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

我的問題是有時候我需要像下面這樣。 第一行是整數n （具有n位數字的1n字符串或數字的長度。

讓我給你舉個例子。

輸入

4
1100

問題是如何限制匹配4位數字（長度）的第二行作為輸入。 怎么樣？ 為什么？

Answer 1

您不能以這種方式限制輸入長度。

按照@Andrey的建議，您應該閱讀輸入，如果輸入不正確，請打印錯誤並閱讀新的輸入。

順便說一下， 在python 3中讀取輸入的pythonic方式是：

a = input("prompt")

因此，您可以執行以下操作：

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