我应该写一个程序,最终可以根据输入内容打印尾巴或头的数量。
我已经编写了程序,当我尝试使用自己的输入时,输出也将正确,但是Codechef抛出了运行时错误(NZEC),我认为这可能是由于使用了较大的内存空间。我在下面粘贴了我的代码,有人可以帮我解决此问题吗?
# cook your dish here
T = int(input())
for i in range(T):
G = int(input())
for j in range(G):
I,N,Q = input().split(" ")
I = int(I)
N = int(N)
Q = int(Q)
coin = []
if I == 1:
coin.extend(['H'] * N)
else:
coin.extend(['T'] * N)
for x in range(N+1):
for y in range(x):
if coin[y] == 'H':
coin[y] = 'T'
elif coin[y] == 'T':
coin[y] = 'H'
ctr = 0
if Q == 1:
for a in coin:
if a == 'H':
ctr += 1
elif Q == 2:
for a in coin:
if a == 'T':
ctr += 1
print(ctr)