测试机python代码IO的改进

Question

我在Codechef做一些练习题。有一个问题叫做Ambiguous permutations：

我的解决办法是：

while 1:
   cnt = int(raw_input())
   if cnt == 0:
      break
  vals = [int(u) for u in raw_input().split(' ')]
  valr = []
  for i in range(cnt):
    valr.append(vals.index(i+1)+1)
  if vals == valr:
    print 'ambiguous'
  else:
    print 'not ambiguous'  

当我在Trypython.org中检查它时，它按预期工作。但当我在Codechef中提交解决方案时，它超时了。你知道吗

我的问题是。代码是否有任何错误（/可以改进），或者是否有任何特定的方法来处理测试机的sysin和输出？你知道吗

[编辑] 接受的解决方案提供了一些很好的建议，我重新考虑了代码逻辑并相应地修改了代码。代码现在可以在时间内运行，尽管它在错误的答案下失败（但是不能在我的测试用例中复制错误的答案）。谢谢你的建议。你知道吗

import sys
def ambigcheck(lis):
   amb = 'ambiguous'
   for i in range(1,len(lis)+1):
       if lis[lis[i-1]-1] != i:
          amb = 'not ambiguous'
          break
   return amb 
while 1:
   cnt = int(sys.stdin.readline())
   if cnt == 0:
       break
   vals = [int(u) for u in sys.stdin.readline().split(' ')]
   sys.stdout.write(ambigcheck(vals))

Answer 1

不要使用raw_input。使用sys.stdin：作为迭代器，或者read()它一次。用sys.stdout.write代替print。尽量只使用一次：预先计算整个输出，然后将其写入屏幕。你知道吗

这将为您提供Python中最快的I/O。你知道吗

更多提示：

• 您不需要构造新的排列，只需检查原始排列中的元素是否有适当的索引即可
• 您不需要执行完全检查，如果至少有一个元素没有正确的索引，则排列并不含糊
• 避免.index()方法，按索引检查，而不是按值检查
• 通过只读取带有itertools.islice的奇数行，您可以更有效地处理输入（在Python中不需要太多的项）