“漂亮数组”CodeChef，如何输入数据

Question

以下是问题陈述：

如果对每一对数字ai，aj，(i≠j)存在ak，使得ak = ai * aj，则数组a称为优美。注意，k也可以等于i或j。 找出给定的数组a是否漂亮！ 输入 输入的第一行包含一个整数T，表示测试用例的数量。测试用例紧随其后。 每个测试用例的第一行包含表示元素数的整数n。 下一行包含n个空格分隔的整数，表示数组a.输出。 对于每个测试用例，输出一行，其中包含“是”或“否”(没有引号)，对应于问题的答案。约束条件 1≤T≤106 1≤n≤105和n超过所有测试用例≤106 -109≤ai≤109 示例 输入 3. 2 0 1 2 1 2 2 5 6 输出： 是 是 不是

我是CodeChef新手，无法正确地接收输入，我有一些代码试图用来解决漂亮的数组问题，但是当我运行它时，我得到了一个NZEC。

from collections import Counter
numCase = int(input())
for i in range(numCase):
    length = input()
    array = Counter(input().split(''))
    answer = "no"
    for i in range(length - 1):
        if array[i] == 0 or array[i] == 1:
            answer = "yes"

    print(answer)

当我更改输入并手动输入代码为我工作的一些数据时，有人能告诉我为什么这段代码不能工作吗？谢谢

Answer 1

我认为你没有检查数组的最后一个元素，因为范围。尝尝这个

for i in range(length)