解析及正确答案:蓝桥村的真相
错误示范:
T=int(input())
m=0
for i in range(T):
n=int(input())
if n%3==0:
m=2*n
print(m)
else:
print(n)
使用此代码提交后,部分用例会出现时间超限错误。
首先,怀疑问题出现在在print中进行计算,新增变量存储计算后的数值,依旧超时。
问题应该出现在多次调用input()和print()导致处理大量数据时速度过慢。
解决方法:
使用一次性读取和输出,使用列表方式存储数据,分别使用下列方式进行输入输出:
sys.stdin.read
sys.stdout.write
最终代码如下:
import sys
input = sys.stdin.read
data = input().split()
T = int(data[0])
results = []
for i in range(1, T + 1):
n = int(data[i])
if n % 3 == 0:
results.append(str(2 * n))
else:
results.append(str(n))
sys.stdout.write("\n".join(results) + "\n")
提交结果:
正确通过。