在特定条件下使用地图功能时停止用户输入

Question

我希望我的代码能够执行此操作：

• 输入的第一行包含一个整数 N ，表示列表中元素的数量;
• 第二行包含以空格分隔的 N 输入。

例如输入

5
1 2 1 1 2 2

由6个数字组成，但应限制为5的输入数。

到目前为止，这是我的代码：

N = int(input(""))
nums = list(map(int, input("").split(" ")))
count, idx, var, index  = 0, 0, 0, 0
for idx, j in enumerate(nums):
    if index == idx:
        var = var + j
        index = var
        if idx == (N-1): continue
        elif j == 0 or j > N:
            count = -1
            break
        else: count += 1
print(count)

Answer 1

据我所知，没有办法限制他们输入的数字，但在你写nums = list(map(int, input("").split(" ")))之后，你可以写nums = nums[0:5]并剪掉任何额外的数字被放入。